pytorch学习
binlin1209
这个作者很懒,什么都没留下…
展开
-
Python中OrderedDict的使用
很多人认为python中的字典是无序的,因为它是按照hash来存储的,但是python中有个模块collections(英文,收集、集合),里面自带了一个子类OrderedDict,实现了对字典对象中元素的排序。请看下面的实例:import collectionsprint "Regular dictionary"d={}d['a']='A'd['b']='B'd['c']='C'for k,v in d.items(): print k,vprint "\nOrder dictio原创 2020-07-16 09:28:38 · 1927 阅读 · 0 评论 -
pytorch学习记录
net = LeNet() # 实例化一个类模型import ipdb; ipdb.set_trace()print(net)net.load_state_dict(torch.load('Lenet.pth')) # 加载模型数据## tensor变量转numpyim = Image.open('cat.jpg')im = transform(im) # [C, H, W]im.numpy() ## tensor变量转numpy ## numpy转tensorsub_ts原创 2020-07-14 08:47:10 · 173 阅读 · 0 评论 -
pytorch torch.nn.AdaptiveAvgPool2d() 自适应平均池化函数解析
如题:只需要给定输出特征图的大小就好,其中通道数前后不发生变化。具体如下:AdaptiveAvgPool2dCLASStorch.nn.AdaptiveAvgPool2d(output_size)[SOURCE]Applies a 2D adaptive average pooling over an input signal composed of several input planes.The output is of size H x W, for any input size. The n原创 2020-07-06 14:58:17 · 5317 阅读 · 0 评论 -
pytorch:model.train和model.eval用法和区别
使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval,eval()时,框架会自动把BN和DropOut固定住,不会取平均,而是用训练好的值,不然的话,一旦test的batch_size过小,很容易就会被BN层导致生成图片颜色失真极大!!!!!!Class Inpaint_Network() ......Model = Inpaint_Nerwoek() #train:Model.train(mode=True).....#test:Model.原创 2020-06-28 09:30:35 · 2030 阅读 · 2 评论