We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
With some error with gpu memory, I've got to relance the train part. In the loading part, a new fc layer is added. What does it mean?
# state_dict = torch.load(args.pretrained)['state_dict'] # from collections import OrderedDict # new_state_dict = OrderedDict() # for k, v in state_dict.items(): # name = k[7:] # new_state_dict[name] = v # model.load_state_dict(new_state_dict) # model.fc = nn.Linear(2048, 80)
The text was updated successfully, but these errors were encountered:
Sorry. I forgot to remove it. It's redundant.
Sorry, something went wrong.
Thanks for your response. By the way, do you have any script to evaluate the trained model on a test set to get mAP?
yea,I have same question ,as we used the caffe version.
No branches or pull requests
With some error with gpu memory, I've got to relance the train part.
In the loading part, a new fc layer is added. What does it mean?
The text was updated successfully, but these errors were encountered: