Yann Henon

Does it work if you keep the line `retinanet = torch.nn.DataParallel(retinanet).cuda()` and pass `CUDA_VISIBLE_DEVICES=0`? It should fall back to the correct behaviour

this is with `retinanet = torch.nn.DataParallel(retinanet).cuda()` ?

@luisriera I believe this is fixed by upgrading to python 3.6 @rvandeghen that seems correct, thanks for your input. If you have time to submit a PR to fix this...

Unfortunately like many personal projects, I just don't have the time needed to dedicate to this project. If there are people interested in being added as contributors so that they...

I use python3, torch 1.4.0, torchvision 0.5.0

Hi, any news on this front? I understand these matters can take time.