- 博客(5)
- 收藏
- 关注
原创 pytorch中Dataloader()中的num_workers设置问题
pytorch中Dataloader()中的num_workers设置问题:如果num_workers的值大于0,要在运行的部分放进__main__()函数里,才不会有错:import numpy as npimport torchfrom torch.autograd import Variableimport torch.nn.functionalimport matplot...
2019-09-20 20:23:31 13836 9
原创 PyTorch官方文档
最近正在学习Pytorch的相关知识,每次查找里面某些函数的参数定义啥的也不太方便,故在这里留下torch的官方文档,以备平时学习总结之需。见:Pytorch英文文档pytorch 中文文档:Pytorch中文文档当然,在pycharm中查找某个函数的具体结构时也可以ctrl+click...
2019-09-19 11:24:35 809
原创 matplotlib中绘制动态图像(实时打印)
使用plt.ion()实时打印拟合过程:import numpy as npimport torchfrom torch.autograd import Variableimport torch.nn.functionalimport matplotlib.pyplot as pltx=torch.unsqueeze(torch.linspace(-1,1,100),dim...
2019-09-19 11:01:31 934
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人