fregu856/deeplabv3

class weights of cityscapes

Chenfeng1271 opened this issue · 3 comments

I don't find the weight file 'class_weights.pkl',can you tell me have to get it

Hi,

You'll have to run utils/preprocess_data.py.

//
Fredrik

Thank u, but I get some new troubles when I train this code directly. It occur 'RuntimeError('cuda runtime error (77) : an illegal memory access was encountered '. So I add 'torch.cuda.set_device(1)' on the top, but here's a new one 'RuntimeError('cuda runtime error (4) : unspecified launch failure at ' These all happen at batch_losses.append(loss_value) in train.py. I try to skip it and sth correspond to it, but the same thing happens in loss.backward(). Could you give me some advice?

Hi,

You'll have to run utils/preprocess_data.py.

// Fredrik
Hi,

The eval code for generating IOU values is missing. Kindly push that code too.