-
Notifications
You must be signed in to change notification settings - Fork 283
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
updated loading pretrained model #49
Conversation
properly raises error when the trained model path is incorrect, prints whether training is starting from scratch or from provided model path.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good and sensible so thanks a lot - just requesting some minor changes.
agent.py
Outdated
elif args.model and not os.path.isfile(args.model): | ||
raise FileNotFoundError(args.model) | ||
|
||
else: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I would delete the else
branch as this is the default and most common behaviour, so there should only be a need to signpost a trained model.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
Co-Authored-By: Kai Arulkumaran <Kaixhin@users.noreply.github.com>
Co-Authored-By: Kai Arulkumaran <Kaixhin@users.noreply.github.com>
Applied your suggestions. Thanks for taking a look |
Ah sorry for not spotting earlier - we can now do |
Np. Done :) |
updated loading pretrained model
properly raises error when the trained model path is incorrect, prints whether training is starting from scratch or from provided model path.