摘要: 使用pytorch的时候,模型训练时,不需要使用forward,只要在实例化一个对象中传入对应的参数就可以自动调用 forward 函数 1 class Module(nn.Module): 2 def __init__(self): 3 super(Module, self).__init__() 阅读全文
posted @ 2024-05-21 15:37 左耳听风 阅读(56) 评论(0) 推荐(0) 编辑