
Pytorch
LeonSun_0101
这个作者很懒,什么都没留下…
展开
-
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 评论 -
Pytorch-tensorboard进阶
Pytorch-tensorboard进阶原创 2018-09-13 20:36:05 · 409 阅读 · 0 评论 -
如何使用特定显卡跑pyTorch
1@https://blog.youkuaiyun.com/u010698086/article/details/80346177方法一:改变系统变量使得仅目标显卡可见export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号,正常的话是从0开始然后再运行pyTorch即可方法二:使用torch.cuda接口#在生成网络对象之前:torch.cuda.se...转载 2018-09-25 22:48:45 · 8332 阅读 · 1 评论 -
【Pytorch】RuntimeError: arguments are located on different GPUs
转载自@https://blog.youkuaiyun.com/okcd00/article/details/784706130x00 前言Pytorch里使用optimizer的时候,由于其会记录step等信息,有时会希望将optimizer的内容记录下来,以备之后继续使用,那么自然而然的会想到使用API中自带的torch.save(object, path)torch.load(path)...转载 2018-10-06 21:22:00 · 5813 阅读 · 2 评论