NTS-Net复现出bug

最近看这篇论文《Learning to Navigate for Fine-grained Classification》，下载了源码想要复现，下载的cuda是9.2版本，python3.7，把鸟类数据集放进去之后，运行train.py,开始报错

top_n_prob = torch.gather(rpn_score, dim=1, index=top_n_index)
RuntimeError: Expected object of scalar type Long but got scalar type Int for argument #3 'index' in call to _th_gather_out

把top_n_index改为long之后，直接退出：Process finished with exit code -1073741819 (0xC0000005)

有无大佬指点一下

NTS-Net复现出bug相关推荐

测试角度的并发和幂等问题总结
本文来自网易云社区作者:刘静媛前言从事金融业务领域的测试工作过程已经有半年时间了,在此期间遇到了各种类型的问题,个人认为在日常测试工作中,特别是涉及资金的业务领域内,测试人员需要格外关注的方 ...
组队瓜分百万奖金池，资深算法工程师带你挑战飞桨论文复现赛！
你是否正在焦虑找不到好的论文? 好不容易找到了paper,无法复现出code? 缺少科研同行交流,只能独自一人闭门造车? 是的,论文复现是要想最快的学习和了解AI领域的方式,复现困境也被叫做" ...
顶尖！百度AI算法专家手把手带你论文复现打比赛！
你是否正在焦虑找不到好的论文? 好不容易找到了paper,无法复现出code? 缺少科研同行交流,只能独自一人闭门造车? 是的,论文复现是要想最快的学习和了解AI领域的方式,复现困境也被叫做" ...
微信重定向地址参数被拦截
这个问题折腾了我和该项目的后台同事好久,记录一下正常流程: 图书馆公众号挂的我们项目的H5链接,公众号菜单点击后先经过后台,后台拿到相关参数跳转到我的页面地址把参数带过来,我获取到参数和用户登录填写 ...
贾扬清开源 AI 框架 Caffe | 开源英雄
[编者按]在开源与人工智能的灿烂星河里,贾扬清的名字都格外地耀眼.因为导师 Trevor Darrell 教授的一句"你是想多花时间写一篇大家估计不是很在意的毕业论文,还是写一个将来大家都会 ...
医学生的编程学习：R Python— — 前言
是否值得掌握一门编程? 这是个数据大爆炸的时代,开源的骑士精神总是不断被提及:这也是个焦虑的时代,知识在网络唾手可及的便利让大家时刻害怕被互联网抛弃.IT成了这个时代的弄潮儿,并渗透到了各个行业,交叉 ...
分析一次安装后EM启动遇到的怪问题
在最初的一次安装Oracle时,应该是Grid.Oracle安装过程不熟悉不够规范,导致了一个非常奇怪的现象. 这种现象描述为以下几点就是 EM使用 emctl start dbconsole命令虽 ...
Nginx一次奇怪的502 报错
1 一次奇怪的线上问题之前遇到nginx报出了如下的error,upstream sent too big header while reading response from upstream,导 ...
一文解决图片数据集太少的问题：详解KerasImageDataAugmentation各参数
作者 | Professor ho 本文转自Professor ho的知乎专栏图像深度学习任务中,面对小数据集,我们往往需要利用Image Data Augmentation图像增广技术来扩充我们的 ...

NTS-Net复现出bug

NTS-Net复现出bug相关推荐

最新文章

热门文章