pre_model is don't work well

[zfs1212]

Thank your code .However,I want to ask some questions.I have download pre-trained model weights into "models" folder then run: python demo.py .But , have error:RuntimeError :cuda runtime error (10): invalid device ordinal at torch/csrc/cuda/module.cpp:32?
I don't know why have error.Can you help me ?
Thx.

[zfs1212]

This issue maybe you use 4GPUs. I retrain model is work.Thx