search

VDIGPKU / T-SEA

[CVPR 2023] T-SEA: Transfer-based Self-Ensemble Attack on Object Detection
93 stars 10 forks source link

检测不到GPU 环境完全是按照requirements.txt里面配的 #22

Open GIANNIS-CHEN opened 5 months ago

GIANNIS-CHEN commented 5 months ago

屏幕截图 2024-06-01 112627

ziyannchen commented 2 months ago

你好，目前我们的代码cpu运行支持得不是很好，这里报错的地方因为我们默认使用了cuda（因为比从cpu搬运到cuda的运行效率更高），可以尝试把报错倒数第二个文件里transformer.py那里.cuda删除即可，也就是直接用torch.FloatTensor()