How to train?

Setup environment

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

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
.idea		.idea
checkpoint		checkpoint
data		data
losses		losses
model		model
tools		tools
DMICC.yaml		DMICC.yaml
README.md		README.md
main.py		main.py