关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>
OpenBMB
/
BMTrain
Not watched
11
10
Fork 17
Code
Releases 0 Wiki Activity
Issues 1 Pull Requests 1 Datasets Model Cloudbrain
Pull Requests
/
Detail
Labels Milestones
New Pull Request

#1 添加 'layer_test'

Open
Lesley_god wants to merge 1 commits from Lesley_god/BMTrain:main into main
Conversation 0 Commits 1 Files Changed 1
Lesley_god commented 1 year ago

def _save_to_state_dict(model : torch.nn.Module, destination, prefix):
if isinstance(model, CheckpointBlock):
if config['rank'] != 0:
destination = OrderedDict() # creates an temporary ordered dict
destination._metadata = OrderedDict()
model.state_dict(destination, prefix, False)
else:
if config['rank'] != 0:
destination = OrderedDict() # creates an temporary ordered dict
destination._metadata = OrderedDict()
model._save_to_state_dict(destination, prefix, False)

def _save_to_state_dict(model : torch.nn.Module, destination, prefix): if isinstance(model, CheckpointBlock): if config['rank'] != 0: destination = OrderedDict() # creates an temporary ordered dict destination._metadata = OrderedDict() model.state_dict(destination, prefix, False) else: if config['rank'] != 0: destination = OrderedDict() # creates an temporary ordered dict destination._metadata = OrderedDict() model._save_to_state_dict(destination, prefix, False)
This pull request can be merged automatically.
You are not authorized to merge this pull request.
Sign in to join this conversation.
Reviewers
Request review
No reviewers
Labels
Apply labels
Clear labels
No items
No Label
Milestone
Set milestone
Clear milestone
No items
No Milestone
Assignees
Assign users
Clear assignees
No Assignees
1 Participants
Notifications
Due Date

No due date set.

Dependencies

This pull request currently doesn't have any dependencies.

Write Preview
Loading…
Cancel
Save
There is no content yet.
Delete Branch 'Lesley_god/BMTrain:main'

Deleting a branch is permanent. It CANNOT be undone. Continue?

No
Yes
Please read the following content carefully:

Dear OpenI User

Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.

For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》

Agree and continue
Disagree, exit