Pytorch model
Pytorch modelmodel.modulesmodel.named_modulesmodel.childrenmodel.named_childrenmodel.parametersmodel.named_parametersmodel.state_dict参考搭建一个简单网络,继承nn.Modulesimport torch import torch.nn as nn class Net(nn.Module): def __init__(self, num_class=10):
原创
2020-12-23 23:02:35 ·
429 阅读 ·
0 评论