2020年9月3日
【小白学PyTorch】5 torchvision预训练模型与数据集全览
摘要： 文章来自：微信公众号【机器学习炼丹术】。一个ai专业研究生的个人学习分享公众号 文章目录： torchvision 官网上的介绍（FQ）：The torchvision package consists of popular datasets, model architectures, and co 阅读全文
posted @ 2020-09-03 13:54 忽逢桃林 阅读(3554) 评论(0) 推荐(0)
【小白学PyTorch】4 构建模型三要素与权重初始化
摘要： 文章目录： 1 模型三要素 三要素其实很简单 必须要继承nn.Module这个类，要让PyTorch知道这个类是一个Module 在__init__(self)中设置好需要的组件，比如conv，pooling，Linear，BatchNorm等等 最后在forward(self,x)中用定义好的组件 阅读全文
posted @ 2020-09-03 00:09 忽逢桃林 阅读(2029) 评论(0) 推荐(0)