peterjc123:

I see that you are missing the if condition protection for the main process. Try again with that and see if that helps.


You mean like this: if cuda: model=nn.DataParallel(model).cuda() ?
Tried. Didn’t help