facebookresearch/ParlAI

Standardize batchifying image features

Open

#2,461 创建于 2020年3月11日

在 GitHub 查看
 (4 评论) (1 反应) (0 负责人)Python (10,371 star) (2,089 fork)batch import
H2 2020Help WantedSmalldonotreap

描述

Currently, ImageSeq2seqAgent and ImagePolyencoderAgent have two separate .batchify_image_features() methods for preparing image features for encoding. The main advantage of ImagePolyencoderAgent's implementation of this method is that the images are packed into a Tensor for data_parallel. It's worth testing whether ImageSeq2seqAgent really requires a separate .batchify_image_features() method now: if not, it should be removed, and ImagePolyencoderAgent.batchify_image_features() should be moved into TorchImageAgent.

贡献者指南