![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
【Pytorch】
甄同学
这个作者很懒,什么都没留下…
展开
-
【Pytorch】建网络的四种方法
转载:https://www.cnblogs.com/denny402/p/7593301.html准备:import torchimport torch.nn.functional as Ffrom collections import OrderedDict方法1:# Method 1 ---------------------------------------...转载 2019-05-08 22:32:22 · 889 阅读 · 0 评论 -
【Pytorch】pytorch模型可视化
step1:转载来自:https://blog.csdn.net/TTdreamloong/article/details/83107110import torchimport torch.nn as nnfrom tensorboardX import SummaryWriterfrom torch.autograd import Variableclass LeNet(nn....转载 2019-05-06 18:46:39 · 4525 阅读 · 0 评论 -
【Pytorch】如何定义nan这个值
>>>import torch>>>n = torch.tensor(float('nan'))>>>ntensor(nan)原创 2019-07-22 10:02:50 · 2863 阅读 · 1 评论 -
【Pytorch】怎么快速的定义cuda版本的变量和网络
torch.set_default_tensor_type('torch.cuda.FloatTensor')此代码可以使得通过torch定义的对象全部为cuda版本.原创 2019-08-29 16:19:31 · 886 阅读 · 0 评论