
深度学习
LeonSun_0101
这个作者很懒,什么都没留下…
展开
-
Pytorch的网络可视化-Tensorboard-pytorch
https://blog.youkuaiyun.com/weixin_40712763/article/details/82292046 安装(点击安装可以查看官方教程) pip install tensorboardX 下载例程 #或者自己建立py文件 mkdir demo cd demo touch demo.py gedit demo.py #然后把下列代码复制进去 import tor...原创 2018-09-01 23:50:24 · 5481 阅读 · 0 评论 -
pytorch 多gpu训练
原文链接overfitover 用nn.DataParallel重新包装一下 数据并行有三种情况 前向过程 device_ids=[0, 1, 2] model = model.cuda(device_ids[0]) model = nn.DataParallel(model, device_ids=device_ids) 只要将model重新包装一下就可以。 后向过程 opt...转载 2019-03-22 20:26:28 · 430 阅读 · 0 评论 -
Pytorch-tensorboard进阶
Pytorch-tensorboard进阶原创 2018-09-13 20:36:05 · 409 阅读 · 0 评论 -
Batch_size
RuntimeError: CUDA error: out of memory 有些时候GPU内存都已经满了,但是GPU的使用百分比还不到百分之10,这是什么情况? https://ptorch.com/news/160.html https://www.zhihu.com/question/32673260...转载 2018-09-09 16:54:13 · 416 阅读 · 0 评论 -
深度学习小技巧
1@开始训练一个模型的时候,在一个很小的数据集上先训练该模型,可以较快地得到大概loss下降的趋势. 2@batch_size的大小会影响优化过程. 3@原创 2018-09-11 23:23:44 · 161 阅读 · 0 评论 -
数据预处理
结合图,很直观@http://7jia.com/1018.html 这一篇很全面@https://blog.youkuaiyun.com/pipisorry/article/details/52247379 数据预处理例子 如下用sigmoid之前需要对数据进行归一化...原创 2018-09-11 14:56:14 · 223 阅读 · 0 评论 -
torch.onnx.export(model, input, "xxx.proto", verbose=True)出现问题
想从pytorch中得到xxx.proto时候发现出现以下问题: Traceback (most recent call last): File "/home/leon/Leon/sparse-to-dense-Leon/main.py", line 357, in <module> main() File "/home/leon/Leon/sparse-to-de...原创 2018-09-02 14:01:42 · 8015 阅读 · 3 评论