Thank you very much!The problem above is solved.But ‘RuntimeError: Expected 4-dimensional input for 4-dimensional weight [24, 3, 3, 3], but got 3-dimensional input of size [3, 3, 158] instead’ error  appears.Thanks a lot!