I would very much like to replicate your work.This error occurred when I started training after loading the dataset successfully.
ValueError: Expected target size (1. 128. 160,960),got torch.size([1. 128.151. 96]),I wonder what caused this error,Thank you for your answer.