06 2021 档案

摘要:CVPR2021文章adco官方代码阅读笔记 1. 正样本相似性计算用的是矩阵乘法,即同一个batch的其它样本也做为负样本(而不仅仅是memory bank中的样本) l_pos = torch.enisum('nc,ck->nk', [q, k.T]) # train.py update_net 阅读全文
posted @ 2021-06-28 10:04 拎壶冲AR 阅读(144) 评论(0) 推荐(0) 编辑
摘要:安装 1. CUDA 路径错误 :/usr/local/cuda/bin/nvcc: not found 莫名其妙多了一个冒号,重新export一下cuda路径 export CUDA_HOME=/usr/local/cuda 2. V0.5用pytorch1.7.1,本地下载git repo,然后 阅读全文
posted @ 2021-06-21 21:36 拎壶冲AR 阅读(1094) 评论(0) 推荐(0) 编辑
摘要:训练模型时,风扇异常响,然后输入nvidia-smi发现风扇异常如下图(网上找的图) 在nvidia论坛有人给出了解决方案,即问题的根源可能是风扇转速不足使GPU过热导致的。 首先开启GPU的persistent mode,再设置风扇的功率,重启即可生效。其中200代表的是风扇的最大功率限制,可以将 阅读全文
posted @ 2021-06-19 19:21 拎壶冲AR 阅读(2472) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示