-
Notifications
You must be signed in to change notification settings - Fork 40
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Running 'run_all_baseline_finetuning.sh': KeyError: 'model' #5
Comments
Hi, what command are you trying to run? |
When I try to do "!. /src/run_all_baseline_finetuning.sh", it prompts me to find it in ". /checkpoints/imagenet" there is no vgg16.pt file needed to run the program, so I downloaded the four model files in advance with the following code and placed them in the imagenet folder. vgg16 = torchvision.models.vgg16(pretrained=True) The four model files were downloaded with torchvision==0.1.9. But I was prompted again with "KeyError: 'model'" |
There is one more question I need to bother you with. README.md says: run_all.sh when training. I understand it as running the following three .sh files. |
Have you replicated successfully this repo, @L-Daniel-ee ? |
Hi @arunmallya :
Running Platform: GoogleColab
Environment: Python3.6
torch==0.2.0.post3
torchvision==0.1.9
torchnet, tqdm
I have changed "raw_input()" to "input()" in network.py. When I try to run the program, the following error occurs.
Traceback (most recent call last):
File "main.py", line 393, in
main()
File "main.py", line 331, in main
model = ckpt['model']
KeyError: 'model'
The file structure below the project is the same as your GitHub commit, where the file structure in checkpoints is shown below.
![image](https://user-images.githubusercontent.com/43088572/137904984-1e6efa01-46fc-4b9c-927b-304d5feb4ba5.png)
The contents of the imagenet folder are shown below.
![image](https://user-images.githubusercontent.com/43088572/137904917-69c123d8-a187-481f-8676-31522c6ed529.png)
The files vgg16.pt, vgg16bn.pt, resnet50.pt, densenet121.pt were downloaded in the same environment with the following code.
vgg16 = torchvision.models.vgg16(pretrained=True)
torch.save(vgg16.state_dict(), "/content/drive/MyDrive/packnet/vgg16.pt").
The other model files are downloaded in the same way.
How should I fix this error? Is there a mistake in the way I download the model? Or is the problem with the network.py file?
One more question.
Is there a requirement for the order in which the .sh files are run? I am talking about the following .sh files.
run_all_baseline_finetuning.sh
run_all_lwf.sh
run_all_sequence.sh
Thanks
The text was updated successfully, but these errors were encountered: