随笔分类 -  Pytorch

Pytorch相关的内容
摘要:几乎所有的常用预训练模型都在这里:https://github.com/pytorch/vision/tree/master/torchvision/models 总结下各种模型的下载地址: ResNet: 'resnet18': 'https://download.pytorch.org/mode 阅读全文
posted @ 2021-09-23 22:44 重大的小鸿 阅读(4551) 评论(0) 推荐(0)
摘要:那么调用forward方法的具体流程是什么样的呢?具体流程是这样的: 以一个Module为例: 1. 调用module的call方法 2. module的call里面调用Module的forward方法 3. forward里面如果碰到module的子类,回到第1步,如果碰到的是Function的子 阅读全文
posted @ 2021-07-28 10:58 重大的小鸿 阅读(390) 评论(0) 推荐(0)