vqafeaturesdataset专题

【数据分析】之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_