Currently the Padding for batch size is done using a torch collate function in ironbeagle-torchmetalBatchMetaCollatePad. Its call hard-codes the largest batch size, 500 currently. Also, the only tested image shape has been concrete through transforms in the helpers.
The desired behavior is to Pad according to the largest batch and largest image efficiently. Finding this across all tasks in an episode needs to be implemented and used in the BatchMetaCollatePad
Currently the Padding for batch size is done using a torch collate function in ironbeagle-torchmetal
BatchMetaCollatePad. Itscallhard-codes the largest batch size,500currently. Also, the only tested image shape has been concrete through transforms in the helpers.The desired behavior is to Pad according to the largest batch and largest image efficiently. Finding this across all tasks in an episode needs to be implemented and used in the
BatchMetaCollatePad