super-gradients/examples/legacy/imagenet_mobilenetv3/mobilenetv3_imagenet_example.py

# MobileNetV3 Large Imagenet classification training:
# This example trains with batch_size = 128 * 2 GPUs, total 256.
# Training time on 2 X GeForce RTX 2080 Ti is 19min / epoch, total time ~ 50 hours.
# Reach 73.79 Top1 accuracy.
# Training parameters are for MobileNet Large

import super_gradients
from omegaconf import DictConfig
import hydra
import pkg_resources


@hydra.main(config_path=pkg_resources.resource_filename("conf", ""), config_name="imagenet_mobilenetv3_conf")
def train(cfg: DictConfig) -> None:
    # INSTANTIATE ALL OBJECTS IN CFG
    cfg = hydra.utils.instantiate(cfg)

    # CONNECT THE DATASET INTERFACE WITH DECI MODEL
    cfg.sg_model.connect_dataset_interface(cfg.dataset_interface, data_loader_num_workers=cfg.data_loader_num_workers)

    # BUILD NETWORK
    cfg.sg_model.build_model(cfg.architecture, load_checkpoint=cfg.load_checkpoint)

    # TRAIN
    cfg.sg_model.train(training_params=cfg.training_params)


if __name__ == "__main__":
    super_gradients.init_trainer()
    train()