Some suggestions about code

[bliblioscar]

Hi. I try to train the model by your command "python -m detector.run --command 'train' --path_to_config ". But there comes some bug when I try to train on gpu(device: 'cuda'). Finally, I fix it by changing some code in "hagrid-master/detector/train.py"

line 128: model.to(conf.device)