facebookresearch/ParlAI

Standardize batchifying image features

Open

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

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

仓库指标

Star
 (10,371 star)
PR 合并指标
 (30 天内没有已合并 PR)

描述

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.

贡献者指南