你好，我在尝试运行该项目的时候（python train.py
--model_name=resnet50
--optim=Adam
--learning_rate=1e-3
--num_epochs=10）
遇到下面这个问题：
Traceback (most recent call last):
File "train.py", line 263, in
main()
File "train.py", line 222, in main
model = get_model(model_weight, use_gpu)
File "/home/gpower/zhangtianjiu/CV/scene-classification/scene-classification97/utils.py", line 46, in get_model
model = model.cuda()
AttributeError: 'dict' object has no attribute 'cuda'
请问应该如何解决呢？非常感谢！
你好，我在尝试运行该项目的时候（python train.py
--model_name=resnet50
--optim=Adam
--learning_rate=1e-3
--num_epochs=10）
遇到下面这个问题：
Traceback (most recent call last):
File "train.py", line 263, in
main()
File "train.py", line 222, in main
model = get_model(model_weight, use_gpu)
File "/home/gpower/zhangtianjiu/CV/scene-classification/scene-classification97/utils.py", line 46, in get_model
model = model.cuda()
AttributeError: 'dict' object has no attribute 'cuda'
请问应该如何解决呢？非常感谢！