摘要: torch.nn 参数Parameter与Module容器 作者:elfin 参考资料来源:torch.nn torch.nn是构建计算图的基础模块,model.train()、model.val()分别为模型的训练、评估模式。 一、Parameter nn.parameter.Parameter( 阅读全文
posted @ 2022-03-31 16:23 巴蜀秀才 阅读(1437) 评论(0) 推荐(0) 编辑