Hi, I run with this line of code in terminal, and I got the following error, could you help me with it?
python train_all.py --model resnet18 --temp trn10
Traceback (most recent call last):
File "train_all.py", line 138, in
main()
File "train_all.py", line 68, in main
small_model = pytorch_resnet.rn_builder(name_to_params[args.model],
KeyError: 'resnet18'
Hi, I run with this line of code in terminal, and I got the following error, could you help me with it?
python train_all.py --model resnet18 --temp trn10
Traceback (most recent call last):
File "train_all.py", line 138, in
main()
File "train_all.py", line 68, in main
small_model = pytorch_resnet.rn_builder(name_to_params[args.model],
KeyError: 'resnet18'