-
Notifications
You must be signed in to change notification settings - Fork 81
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
zero doc & checkpoint save adapt #834
Conversation
@property | ||
def bias(self): | ||
return self.net.bias | ||
|
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.
may change if self.net.has_bias
-> if self.bias is not None
since mint.Conv does not have has_bias
attribute
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.
ok
mindone/trainers/callback.py
Outdated
if self.ckpt_combine_online: | ||
new_net_to_save = self._do_ckpt_combine_online() | ||
|
||
if self.self.need_save_network: |
This comment was marked as resolved.
This comment was marked as resolved.
Sorry, something went wrong.
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
* ckpt save when zero * add docs * fix bugs * op_group->optimizer_parallel_group * ms_class -> jit_class * transform_checkpoints->convert_checkpoints * typo * fix bugs --------- Co-authored-by: zhaoting <zhaoting23@huawei.com>
* ckpt save when zero * add docs * fix bugs * op_group->optimizer_parallel_group * ms_class -> jit_class * transform_checkpoints->convert_checkpoints * typo * fix bugs --------- Co-authored-by: zhaoting <zhaoting23@huawei.com>
What does this PR do?
Fixes # (issue)
Adds # (feature)
Before submitting
What's New
. Here are thedocumentation guidelines
Who can review?
Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.
@xxx