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_
相關文章
相關標籤/搜索