facebookresearch/ParlAI

Standardize batchifying image features

Open

#2,461 opened on 2020年3月11日

GitHub で見る
 (4 comments) (1 reaction) (0 assignees)Python (10,371 stars) (2,089 forks)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.

コントリビューターガイド

Standardize batchifying image features · facebookresearch/ParlAI#2461 | Good First Issue