-
Notifications
You must be signed in to change notification settings - Fork 176
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
Saving model after finetuning #33
Comments
you gotta edit a line in train.py. In the PyTorch scripts, apparently the block that calls you can add this line
|
thx. this saved as a tar file, which contains a set of files. should it be a simple pt file? |
I did as skcerin said. But failed to load to train, it says "expected cpu but got cuda:0" |
how can use it? for test |
How do I save the model after fine-tuning using command line on dataset(CIFAR100) ?
The text was updated successfully, but these errors were encountered: