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 · 273 阅读 · 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 dsimpo...原创 2018-03-15 21:24:28 · 16148 阅读 · 2 评论 -
PyTorch Variable与动态图
https://mp.weixin.qq.com/s/OMjfck4jlMneGZ1NJxbjKQfor data, label in trainloader: ...... out = model(data) loss = criterion(out, label) loss_sum += loss # <--- 这里 .........转载 2018-03-20 08:58:15 · 1269 阅读 · 0 评论 -
PyTorch 测试阶段显存爆炸的一个解决方法
with torch.no_grad(): output = self.model(lr)原创 2019-04-04 19:32:40 · 3284 阅读 · 0 评论