随笔分类 -  Pytorch

Demo——Image classification
摘要：使用pytorch。 一、定义模型结构 model.py import torch.nn as nn import torch.nn.functional as F """ 定义一个类，这个类继承于nn.Module，实现两个方法：初始化函数和正向传播 实例化这个类之后，将参数传入这个类中，进行正向 阅读全文
posted @ 2020-03-13 14:37 Ariel_一只猫的旅行 阅读(269) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示