facebookresearch/ParlAI

Standardize batchifying image features

開放

#2,461 建立於 2020年3月11日

 (4 則留言) (1 個反應) (0 位負責人)Python (2,089 個分叉)batch import
H2 2020Help WantedSmalldonotreap

倉庫指標

星標
 (10,371 顆星)
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.

貢獻者指南