search

MegEngine / Models

采用MegEngine实现的各种主流深度学习模型
Other
303 stars 99 forks source link

fix(classification): only save state_dict at rank 0 #97

Closed zhouyizhuang-megvii closed 3 years ago

wjfwzzc commented 3 years ago

LGTM