![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
钟良建呀
计算机视觉
展开
-
pytroch-自定义模块
自定义模块 1.自定义梯度求导函数(custom autograd function) 主要用于自定义函数的自动求导 2.自定义网络模型(custom model) 主要用于复杂网络模型的构建 """自定义模块 1.自定义梯度求导函数(custom autograd function) 定义:首先继承torch.autograd.Functionl类,然后重写forward和backward函数 使用:调用该类的apply方法 """ # define custom autograd_function im原创 2020-08-24 12:49:44 · 228 阅读 · 0 评论 -
pytorch-网络模型定义
网络模型定义 构建网络模型类 1.该类必须继承torch.nn.Module,并且在初始化方法中引入父类初始化方法 2.在__init__()中定义并初始化网络 3.在forward()中定义前向传递(反向传播根据前向传递自动定义) 使用时直接初始化该类,传入输入即可的得到输出 """模块导入""" import torch import torch.nn as nn # 包含模型设计的模块和类 import torch.nn.functional as F """网络模型定义""" class Net原创 2020-08-22 21:36:15 · 434 阅读 · 0 评论 -
pytorch-模型保存与加载
模型保存/加载的四种方法 1.保存/加载状态字典(state_dict) 2.保存/加载整个模型(entire model) 3.保存/加载checkpoint信息 4.保存/加载多个模型到一个文件 注:详情请参阅 pytorch 官方文档 添加链接描述 """模型保存与加载 方法1:保存/加载状态字典(state_dict) 该方法具有更大的灵活性,推荐使用 方法2:保存/加载整个模型(entire model) 方法3:保存/加载checkpoint 该方法以字典形式存储模型信息,推荐训练过程使用原创 2020-08-22 19:46:00 · 232 阅读 · 0 评论