
pytorch
XianyuFFF
Stay hungry Stay foolish
展开
-
使用Detectron2训练自己的网络 (一)
首先上官方文档,本文希望能给看文档比较费劲的萌新一些帮助。https://detectron2.readthedocs.io/tutorials/datasets.html(一)使用自定义数据集【1】网上很多教程是将数据集转化为coco数据集格式,然后直接使用detectron2自带的方法解析数据集。coco解析数据集的方法位置:detectron2/data/datasets/coco.py的load_coco_json。该种方法我就不做过多介绍了,网上很多。【2】自己定义解析数据原创 2020-05-10 21:25:46 · 4092 阅读 · 5 评论 -
RuntimeError: cuda runtime error : device-side assert triggered
这个错误整了我一天时间,所以特此记录一下,如果大家也碰到希望能够避雷!!!这个错误是在运行PyTorch代码出错,我碰到的原因是神经网络init定义时给的参数出错。(具体是nn.Embedding(vocab_size,embed_dim)中vocab_size传错参数导致的)【1】网上有人说这是gpu坏了,不能运行,首先可以通过:import torchtorch.cuda.is_available()测试,就可以排除这种问题。所一般不是cuda或者机器的问题注:对于报错,不要原创 2020-05-10 20:05:07 · 3037 阅读 · 2 评论 -
Pytorch使用技巧(一)
1.pytorch使用GPU资源时需要显示调用,如果不指定的话,默认是使用cpu计算资源input=input.to(device)注意:一定要赋值于一个变量2.torch随机矩阵:a=torch.randn(1,3,3)#生成1*3*3的随机矩阵#tensorflow中的随机矩阵b=tf.random_uniform([1,3,3])3.在使用 torch.m...原创 2019-11-24 20:51:27 · 316 阅读 · 0 评论