I’m newbie to pytorch. i’m working currently on untrimmed video classification. the network i’m implementing does online learning which means a batch size of 1. In addition, the loss is calculated on all frames of a single video (the loss is equal to the mean of cross entropy across all frames).
the tutorial : DataParallel — PyTorch 1.8.1 documentation was not
of a great help since my batch size is 1 so the network runs only on GPU 1.