Fixed multi-GPU training#6

Merged

dnth merged 5 commits intodnth:mainfrom

EnriqueGlv:deimkit

Mar 21, 2025

Commits on Mar 14, 2025

Added torchvision.transforms.v2.Transform.transform overrides to enable torchvision>=0.21 support
EnriqueGlv
committed

Commits on Mar 17, 2025

Merge remote-tracking branch 'deimkit/main' into deimkit
EnriqueGlv
committed

Commits on Mar 18, 2025

Fixed distributed training for Multi-GPU support
EnriqueGlv
committed

Commits on Mar 19, 2025