working on google colab

Does the code work on google colab ?

I have this error “KeyError: Caught KeyError in DataLoader worker process 0.” when calling train(segmentation_module, iterator_train, optimizers, history, epoch+1, cfg) after running the command python3 train.py --gpu 1 …

[xksteven]

train.py needs to be modified as per this issue to run on a single GPU.

to run on a single GPU it’s

train.py --gpu 0 

as opposed to train.py --gpu 1 .

I’ve also uploaded the pretrained models that can be found here

Finally to run the eval_ood.py if using the default config/ade20k-resnet50dilated-ppm_deepsup.yaml and the model I’ve uploaded, change the num_classes parameter in the yaml file to be 13.

Thanks for pointing out the typo. I fixed it and another as well.

I’ve also updated the README as well to reflect these changes

[julienguegan]

Thanks for the update @xksteven . The previous error have disappeared but I think there is still something missing because I can’t make the command python train.py works… I had to add feed_dict = feed_dict[0] in line 32 of models.py because I had a list and not a dictionnary but then I have “Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same”. You can check link_colab if you have time

PS : I also noted one minor error in eval_ood.py : in_scores = - conf[np.logical_not(out_label))] line 51, there is one parenthesis too many