batch_size=2,training slows down. #60
Description
The log generates warnings, and training also slows down. Training on videos of the same duration is faster when there is no batch size of 1.
UserWarning: Using a target size (torch.Size([1, 3, 512, 512])) that is different to the input size (torch.Size([2, 3, 512, 512])). This will likely lead to incorrect results due to broadcasting. Please ensure they have the same size.