MobileViT/tests

• modules for unit test (UT): test the main modules including dataset, transform, loader, model, loss, optimizer, and scheduler.

To test all modules:

pytest tests/modules/*.py

• tasks for system test (ST): test the training and validation pipeline.

To test the training process (in graph mode and pynative+ms_function mode) and the validation process, run

pytest tests/tasks/test_train_val_imagenet_subset.py

To test training in distributed mode if you have multiple GPUs:

pytest tests/tasks/parallel/test_parallel_train_val.py