2020年11月3日
pytorch 参数初始化
摘要： 利用pytorch 定义自己的网络模型时，需要继承toch.nn.Module 基类。 基类中有parameters()、modules()、children()等方法 import torch import torch.nn as nn class myModel(nn.Module): def 阅读全文
posted @ 2020-11-03 22:03 learningcaiji 阅读(763) 评论(0) 推荐(0) 编辑