karthikramesh55:

Are there any ways that you would suggest adding the augmented tensors to the training set?


You could create two separate datasets, the standard one and the other one with transformations, and combine them into a ConcatDataset.



 karthikramesh55:

Also, is it possible to add the transformation on the existing training data on the fly (dynamically) while training the model?


The transformations will be applied on-the-fly by default on each sample in the __getitem__ method of the dataset.