摘要:
文章来自:微信公众号【机器学习炼丹术】。一个ai专业研究生的个人学习分享公众号 文章目录: torchvision 官网上的介绍(FQ):The torchvision package consists of popular datasets, model architectures, and co 阅读全文
摘要:
文章目录: 1 模型三要素 三要素其实很简单 必须要继承nn.Module这个类,要让PyTorch知道这个类是一个Module 在__init__(self)中设置好需要的组件,比如conv,pooling,Linear,BatchNorm等等 最后在forward(self,x)中用定义好的组件 阅读全文