Hello, I think your server does not support a GPU, even using "gpu": true
in aicrowd.json. I’m using pytorch==1.2 I checked two scenarios:
- create the model, and call
model.cuda()
- the evaluation stops after the first iteration. Create the model without moving it to cuda(using cpu) - it works fine. - create the model and load locally pre-trained weights(on the GPU) - the evaluation stops after the first iteration. I think this is also due to the lack of a GPU in your environment. In my local environment with the same docker image everything works fine.