Install packages as: pip install DMICC.yaml
To train DMICC, we take main.py as an example, run the following commands sequentially to perform our method on STL-10:
python main.py -g 0,1,2,3 -n 8 --batch_size 128 --epochs 5000 --pretrained checkpoint_dir/checkpoint.pth