-
Notifications
You must be signed in to change notification settings - Fork 50
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
558 implement a base class for scripting models #559
558 implement a base class for scripting models #559
Conversation
config/gdl_config_template.yaml
Outdated
@@ -34,7 +34,7 @@ general: | |||
raw_data_dir: data | |||
raw_data_csv: tests/tiling/tiling_segmentation_binary_ci.csv | |||
tiling_data_dir: ${general.raw_data_dir}/patches # where the patches will be saved | |||
save_weights_dir: saved_model/${general.project_name} |
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.
why changing the template?
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 changed it because the default ParentPath is set to the current working directory which might not be desirable.
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.
Change reverted beause it fails tests!
The merge-base changed after approval.
@@ -0,0 +1,32 @@ | |||
import torch | |||
|
|||
class ScriptModel(torch.nn.Module): |
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.
Hey! Very curious to start using scripted models. Thanks for this addition. Could a test for the __init__ and forward() to validate these methods act as expected be relevant?
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.
yes, I added tests.
Thank you for submitting your PR and helping us with GDL.
Please read the template below, fill it out as appropriate, and make additional changes to your code as needed.
Please feel free to submit your PR even if it doesn't satisfy all of the requirements below.
If you need assistance or review from a maintainer, add the label Status: Help Needed or Status: Review Needed respectively.
After review, a maintainer will add the label Status: Revision Needed if further work is required for the PR to be merged.
Description
Please include a summary of the change and which issue is resolved (if applicable).
Please also include relevant motivation and context.
List any dependencies that are required for this change.
If it fixes an open issue, please link to the issue here.
Fixes #558
Type of change
What types of changes does your code introduce? Put an
x
in all the boxes that apply:How Has This Been Tested?
If the added changes are already cover by tests, skip that section.
Otherwise, please describe tests that you added to the pytest codebase (if applicable).
Screenshots (if applicable):
Checklist:
If you're unsure about any of these, don't hesitate to ask. We're here to help!
If your PR does not fulfill all of the requirements in the checklist above, that's OK! If you need help, @-mention a maintainer and/or add the Status: Help Needed label.