![](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_functionim原创 2020-08-24 12:49:44 · 246 阅读 · 0 评论 -
pytorch-网络模型定义
网络模型定义构建网络模型类1.该类必须继承torch.nn.Module,并且在初始化方法中引入父类初始化方法2.在__init__()中定义并初始化网络3.在forward()中定义前向传递(反向传播根据前向传递自动定义)使用时直接初始化该类,传入输入即可的得到输出"""模块导入"""import torchimport torch.nn as nn # 包含模型设计的模块和类import torch.nn.functional as F """网络模型定义"""class Net原创 2020-08-22 21:36:15 · 458 阅读 · 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 · 243 阅读 · 0 评论