A script for training the ConvNextV2 on CIFAR10 dataset using the FSDP technique for a distributed training scheme.
deep-learning
distributed-computing
pytorch
distributed
distributed-training
distributed-learning
convnext
convnextv2
fsdp
fully-sharded-data-parallel
-
Updated
Dec 11, 2023 - Python