regat专题

VQA-ReGat 项目运行遇到的错误

VQA-ReGat:关系感知图形注意网络用于VQA 项目地址 论文地址 1.torch报错:StopIteration: Caught StopIteration in replica 0 on device 0. 原因:多GPU运行此项目报错,可能是torch版本错误。 修改:按照别的博客将 weight = next(self.parameters()).data改为weight = t

【数据分析】之ReGat的VQAFeaturesDataset加载

1 .VQAFeatureDataset 此类是ReGat项目对torch自带的from torch.utils.data import Dataset的重写,是模型运行的时候训练集和测试集的加载,加载的数据是模型forward函数的参数。如下: regat.forward(): def forward(self, v, b, q, implicit_pos_emb, sem_adj_