@ptrblck Thank you so much , now it is breaks on 5 th epoch with error
“RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 1, 3, 3], but got 5-dimensional input of size [1, 1, 512, 512, 3] instead”