2020年9月2日
常用的神经网络模型和pytorch实现(一)
摘要： BasicModule 程序实现的时候所有模型继承自定义的basicmoudle，主要重写了模型加载和保存等方法 1 import torch 2 from config import opt 3 class BasicModule(torch.nn.Module): 4 def __init__( 阅读全文
posted @ 2020-09-02 12:06 你的雷哥 阅读(2105) 评论(0) 推荐(0) 编辑