pytorch加载state_dict()保存的权重

import torch from mobilenetv3 import MobileNetV3_Large from mobilenetv3 import MobileNetV3_Small import collections if __name__ == '__main__': dic = collections.OrderedDict() net=MobileNetV3_
相关文章
相关标签/搜索