首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
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_
阅读更多...