- 
                Notifications
    You must be signed in to change notification settings 
- Fork 25.7k
[PyTorch Edge] Set training for module only #89488
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
Conversation
Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/) [ghstack-poisoned]
| 🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/89488
 Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit 03d89cd: This comment was automatically generated by Dr. CI and updates every 15 minutes. | 
Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/) ghstack-source-id: 174195645 Pull Request resolved: #89488
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.
LGTM
Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/) [ghstack-poisoned]
Pull Request resolved: #89488 Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. ghstack-source-id: 174243857 Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/)
Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/) [ghstack-poisoned]
Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/) [ghstack-poisoned]
Pull Request resolved: #89488 Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. ghstack-source-id: 174538868 Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/)
| @pytorchbot merge (Initiating merge automatically since Phabricator Diff has merged) | 
| Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team | 
Update previous recursive logic. Continue setting training attribute only if the slot is an object and a module. For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively. This change patch one fix to set training attribute for `model_f269583363.ptl`. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate. Differential Revision: [D41466417](https://our.internmc.facebook.com/intern/diff/D41466417/) Pull Request resolved: pytorch#89488 Approved by: https://github.com/iseeyuan
Stack from ghstack (oldest at bottom):
Update previous recursive logic.
Continue setting training attribute only if the slot is an object and a module.
For the corresponding JIT module, they get the module list first and set module one by one. there is method to get all modules iteratively, instead of recursively.
This change patch one fix to set training attribute for
model_f269583363.ptl. Another patch is needed, because current lite interpreter doesn't have the correct type when loading object with setstate.Differential Revision: D41466417