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.

貢獻者指南