GAIA-vision / GAIA-cls

GAIA-cls

An AutoML toolbox specialized in classification.

Install

requirements:

torch 1.9.0+

gaiavision

mmcv-full 1.3.9

Command

Supernet training

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 bash scripts/train_local.sh ./configs/conformer/tiny/Auto-Conformer_tiny_layer_15_train_noaug.py 8