Train batch generic#724

felipemello1 merged 8 commits into

meta-pytorch:mainmeta-pytorch/torchforge:mainfrom

HosseinKaviani-H:TrainBatch_GenericHosseinKaviani-H/forge:TrainBatch_GenericCopy head branch name to clipboard