@ptrblck I want to physically increase the number of images as stated in the paper because the data set is imabalnaced (class1: 2555, class 2: 227, class 3: 621, class 4: 2552 images). If I increase the epochs, transforms will be applied equally to all of them. And the classification will be based on the majority class.