运行PRDH遇到的cuda错误

[yaoguangzi]

请问一下您遇见过这个问题吗？ 非常感谢 File "deep-cross-modal-hashing-master/torchcmh/training/PRDH.py", line 342, in train trainer.train() File "deep-cross-modal-hashing-master/torchcmh/training/PRDH.py", line 110, in train self.valid(epoch) File "deep-cross-modal-hashing-master/torchcmh/training/base.py", line 162, in valid self.valid_calc(self.img_model, self.txt_model, self.valid_data, self.bit, self.batch_size, return_hash=True, cuda=self.cuda) File "deep-cross-modal-hashing-master/torchcmh/training/base.py", line 206, in valid_calc rB_img, rB_txt = TrainBase.get_codes(img_model, txt_model, dataset, bit, batch_size, cuda=cuda) File "deep-cross-modal-hashing-master/torchcmh/training/base.py", line 244, in get_codes img_buffer[index, :] = img_hash.data

RuntimeError: CUDA error: an illegal memory access was encountered

[WangGodder]

检查代码是否img_hash和img_buffer都是cuda数据，如果不是就改掉

[yaoguangzi]

非常感谢您及时回复。在未改上述IMG_BUFFER的另一次运行中，也是报该错误。且在DCMH和CMHH中没有遇见过该错，您觉得问题出现在了哪里

File "main.py", line 21, in run() File "/home/jrr/deep-cross-modal-hashing-master/torchcmh/run.py", line 41, in run get_train(method, data_name.upper(), img_dir, bit, batch_size=batch_size, cuda=cuda, kwargs) File "/home/jrr/deep-cross-modal-hashing-master/torchcmh/run.py", line 55, in get_train train(dataset_name, img_dir, bit, kwargs) File "/home/jrr/deep-cross-modal-hashing-master/torchcmh/training/PRDH.py", line 342, in train trainer.train() File "/home/jrr/deep-cross-modal-hashing-master/torchcmh/training/PRDH.py", line 79, in train self.remark_loss(intra_loss, intra_loss, decorrelation_loss, reg_loss, loss) File "/home/jrr/deep-cross-modal-hashing-master/torchcmh/training/base.py", line 115, in remark_loss self.loss_store[loss_name].update(args[i].item()) RuntimeError: CUDA error: an illegal memory access was encountered

[WangGodder]

几年前写的代码了，忘记了，你参考DCMH的代码review下吧。

[yaoguangzi]

非常感谢~ 好滴~