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