Python
qq_34535410
这个作者很懒,什么都没留下…
展开
-
Python yield生成器
Python 中可以通过列表生成式,创建列表,并且可以通过列表迭代访问列表里的每一个元素,如果元素数量很大,而且只需访问前几个元素,使用列表将会造成很大的浪费,生成器就是为了解决这种问题而设计的。 创建generator的方法 1. 把[]改成() alist = [x*x for x in range(5)] print(alist) blist = (x*x for x i原创 2018-02-01 22:08:49 · 302 阅读 · 0 评论 -
Pytorch读取图片并显示
# -*- coding: utf-8 -*- # @Time : 18-3-15 下午6:43 # @Author : zhwzhong # @File : model.py # @Contact : zhwzhong.hit@gmail.com # @Function: from torchvision import transforms, datasets as ds impo...原创 2018-03-15 21:24:28 · 16193 阅读 · 2 评论 -
PyTorch Variable与动态图
https://mp.weixin.qq.com/s/OMjfck4jlMneGZ1NJxbjKQ for data, label in trainloader: ...... out = model(data) loss = criterion(out, label) loss_sum += loss # <--- 这里 ...... ...转载 2018-03-20 08:58:15 · 1296 阅读 · 0 评论 -
PyTorch 测试阶段显存爆炸的一个解决方法
with torch.no_grad(): output = self.model(lr)原创 2019-04-04 19:32:40 · 3313 阅读 · 0 评论