2020年9月5日
基于pytorch神经网络模型参数的加载及自定义
摘要： 最近在训练MobileNet时经常会对其模型参数进行各种操作，或者替换其中的几层之类的，故总结一下用到的对神经网络参数的各种操作方法。 1.将matlab的.mat格式参数整理转换为tensor类型的模型参数 import torch import torch.nn as nn import tor 阅读全文
posted @ 2020-09-05 11:38 爱吃虾的小暹罗 阅读(1633) 评论(0) 推荐(0) 编辑