Bug: Conditional transformation of data_loader

[GeeekySouro003]

Conditional Transformation: The get_transforms() function is called twice, once in get_data_loaders() for training and again (potentially) for validation. You can potentially define the transformation logic within get_data_loaders() and conditionally apply normalization based on whether it's training or validation data. This could reduce code duplication.

[sagar132103]

I want to work on this.

[GeeekySouro003]

@sagar132103 assigned