pytorch
onion_rain
这个作者很懒,什么都没留下…
展开
-
torch.autograd.set_detect_anomaly在mmdetection中的用法
torch.autograd.set_detect_anomaly原创 2023-01-31 13:41:21 · 577 阅读 · 0 评论 -
list、numpy、tensor(pytorch)常用操作
numpy 2 tensort = torch.from_numpy(n)tensor 2 numpyn = t.numpy()原创 2020-07-20 11:56:35 · 2941 阅读 · 2 评论 -
vscode local attach debug配置与使用
配置debug用vscode打开pytorch源码,run侧边栏中新建launch.json选择attach using process ID类型 的上面那个remote attach(因为我尝试了attach process id总是搞不定。。)配置好port和host,再在python代码中加入此时执行便会等待attach:(pytorch) xueruini@nico2:~/onion_rain/pytorch/object_detection$ python train.py --j原创 2020-07-09 15:47:16 · 3362 阅读 · 0 评论 -
多卡训练报错:assert all(map(lambda i: i.is_cuda, inputs))
由于debug需要管理员权限。。未完待续。。。起因多卡训练模型时候出错:training...原创 2020-07-08 20:10:57 · 1212 阅读 · 2 评论 -
编译pytorch源码
编译pytorch源码clone pytorch源码:(base) xueruini@nico2:~/onion_rain/pytorch/code$ git clone --recursive https://github.com/pytorch/pytorchCloning into 'pytorch'...remote: Enumerating objects: 37, done.remote: Counting objects: 100% (37/37), done....Resolv原创 2020-07-03 18:42:19 · 1731 阅读 · 3 评论 -
linux下批量清理占用显存的进程
在跑TensorFlow、pytorch之类的需要CUDA的程序时,强行Kill掉进程后发现显存仍然占用,这时候可以使用如下命令查看到top或者ps中看不到的进程,之后再kill掉:fuser -v /dev/nvidia*批量清理显卡中残留进程(无sudo权限):fuser -v /dev/nvidia* |awk '{for(i=1;i<=NF;i++)print "kill -...转载 2020-03-21 20:52:07 · 1410 阅读 · 0 评论 -
intel神经网络压缩库distiller使用时遇到的问题
ModuleNotFoundError: No module named ‘distiller’原因:一开始没有按照README给出的方法,用conda新建了虚拟环境然后使用pip install -r requirements.txt安装好了运行出错才发现。按照README提示pip3 install -e .装好之后还是没有解决上述报错,无奈删除虚拟环境,重建虚拟环境重新执行安装...原创 2020-03-04 23:09:45 · 609 阅读 · 0 评论 -
imagenet2012训练集警告:UserWarning: Corrupt EXIF data. Expecting to read 4 bytes but only got 0. warni
n04152593_17460.JPEG 其实是 HEIC 图像更多imagenet训练集特殊图像原创 2020-02-23 18:45:38 · 3167 阅读 · 2 评论 -
卷积、池化输出尺寸计算
原文链接通常情况下padding[0] = padding[1]dilation[0] = dilation[1] = 1kernel_size[0] = kernel_size[1]Hin = WinHout = Wout简化为Hout = Wout = (Hin + 2*padding - 1)/stride + 1(向下取整)...原创 2020-02-11 12:07:25 · 768 阅读 · 0 评论