pytorch
JY丫丫
这个作者很懒,什么都没留下…
展开
-
pytorch 一些基础函数
torch.zeros_like(input)返回跟input的tensor一个size的全零tensor原创 2020-08-11 14:58:37 · 216 阅读 · 0 评论 -
NotImplementedError: Got <class ‘__main__.AverageMeter‘>, but expected numpy array or torch tensor
问题 File "main_finetune_pointConv_torchModel.py", line 484, in <module> main() File "main_finetune_pointConv_torchModel.py", line 296, in main train(train_loader, model, criterion, optimizer, epoch) File "main_finetune_pointConv_torchMod原创 2020-08-04 16:53:23 · 3523 阅读 · 2 评论 -
RuntimeError: CUDA out of memory. Tried to allocate 30.00 MiB
RuntimeError: CUDA out of memory. Tried to allocate 30.00 MiB (GPU 0; 10.92 GiB total capacity; 9.65 GiB already allocated; 29.00 MiB free; 10.37 GiB reserved in total by PyTorch)GPU跑模型报错RuntimeError: CUDA out of memory. Tried to allocate 38.00 MiB (GPU原创 2020-06-19 15:13:25 · 10890 阅读 · 0 评论 -
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
文章目录问题原因解决方式问题----------one epoch begin----------the compression rate now is 0.700000pruning_cifar10_resnet.py:302: UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad...原创 2020-03-31 19:02:20 · 2655 阅读 · 0 评论 -
ValueError: Image size of 1200x102500 pixels is too large. It must be less than 2^16 in each directi
文章目录问题情况原因解决方式问题情况Traceback (most recent call last): File "CNNVisualisation.py", line 335, in <module> plot_weights(alexnet, 3, single_channel = True, saveName='alexnet, 3') File "CNNV...原创 2020-03-21 14:35:18 · 11753 阅读 · 0 评论 -
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is Fal
RuntimeError: Attempting to deserialize object on a CUDA device buttorch.cuda.is_available() is False. If you are running on a CPU-onlymachine, please use torch.load with map_location=torch.device(...原创 2020-03-20 23:51:38 · 983 阅读 · 0 评论 -
ValueError: setting an array element with a sequence
ValueError: setting an array element with a sequence问题主要有几种以下的情况:输入到模型里的数据没严格转换成 np.arry()形式,矩阵没对齐,个别行成员数量与其他不一致,不如0的缺省等。数据dtype不对或者说不一致,应该统一成 np.float64, int或者其他。如feeds = { sbl.input_x1: x...原创 2020-03-13 20:24:35 · 2937 阅读 · 0 评论 -
Pytorch设置多线程进行dataloader
使用 GPU 进行训练的时候,torch.utils.data.DataLoader中的 num_workers 设置为0,设pytorch 自已决定参考文献(1条消息)Pytorch设置多线程进行dataloader时影响GPU运行_xiuxiuxiuxiul的博客-CSDN博客https://blog.csdn.net/xiuxiuxiuxiul/article/details/...原创 2020-01-03 17:00:22 · 4062 阅读 · 0 评论 -
No module named 'tensorboardX'
报错 No module named ‘tensorboardX’pip install tensorboardX原创 2019-12-11 14:17:59 · 14257 阅读 · 0 评论 -
Expected object of scalar type Float but got scalar type Double for argument
引言该问题的原因是,数据的类型不匹配所导致的解决方式更改数据类型如:X_trainTensor–> X_trainTensor.float()When the error is RuntimeError: Expected object of scalar type Float but got scalar type Double for argument #4...原创 2019-10-28 14:37:01 · 2195 阅读 · 0 评论 -
(转)RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.DoubleTensor) should be
出错位置: train_label_batch = torch.from_numpy(train_label_batch)解决办法:将数据类型转换成FloatTensor即可,如下,加一行代码Tensor.type(torch.FloatTensor)train_label_batch = t...转载 2019-10-28 14:26:02 · 1435 阅读 · 3 评论 -
Pytorch使用manual_seed(设置随机数生成种子)
设置随机种子,保证每一次生成的随机数都是一样的import torchtorch.manual_seed(2) #为CPU设置种子用于生成随机数,以使得结果是确定的 print(torch.rand(2))if args.cuda: torch.cuda.manual_seed(args.seed) #为当前GPU设置随机种子; # 如果使用多个GPU,应该使用...原创 2019-10-16 10:04:39 · 3256 阅读 · 0 评论 -
PyTorch中指定GPU 运行
pytorch 进行深度学习训练模型服务器上有多个GPU,需要制定代码在其中某个GPU运行时,可以采用下面的方式进行指定目录1、查看服务器上的gpu信息2、获取id,这里取的是id 13、在代码的开始位置添加以下的代码参考文献查看指定GPU的id,见以下1、查看服务器上的gpu信息服务器上的终端上输入watch -n 0.5 nvid...原创 2019-08-16 10:17:07 · 8141 阅读 · 0 评论 -
本地电脑连接服务器tensorboard
前言在使用tensorboard的时候,会遇到这样的一个问题,实验的结果在服务器上,但想在不下载『实验结果〗的前提下想在本地上查看结果。该文章记录的解决这样的一个问题适用: tensorflow、pytorch需要软件:xshell ...转载 2019-08-21 11:10:27 · 1537 阅读 · 0 评论 -
pytorch 中的 forward 的使用与解释
文章目录前言forward 的使用forward 使用的解释参考资料前言最近在使用pytorch的时候,模型训练时,不需要使用forward,只要在实例化一个对象中传入对应的参数就可以自动调用 forward 函数即:forward 的使用class Module(nn.Module): def __init__(self): super(Module, se...原创 2019-07-25 17:58:29 · 80089 阅读 · 11 评论 -
win7 系统下pytorch安装
系统环境介绍:win7 系统python 3.6python 2.7tensorflowAnaconda3 (64-bit)步聚1、win7 -》 开始 --> 运行 -->win7运行终端2、输入以下命令:torch、torchvision 安装pip3 install torch或者是pip3 ins...原创 2019-06-26 15:38:55 · 2365 阅读 · 0 评论