CIFAR10 TorchPie

Requirements

Install TorchPie

pip install git+https://git.dev.tencent.com/SunDoge/torchpie.git

Use seleted GPUs

python -m torchpie.distributed.launch --gpus 0,1 main.py -c config/default.conf -e exps/resnet20-2gpu

Use all GPUs

python -m torchpie.distributed.launch main.py -c config/default.conf -e exps/resnet20-2gpu

Make sure you use different experiment-path every time you run.

ln -s ../torchpie/torchpie ./

Name		Name	Last commit message	Last commit date
Latest commit History 28 Commits
config		config
datasets		datasets
models		models
scripts		scripts
utils		utils
.gitignore		.gitignore
README.md		README.md
main.py		main.py
transforms.ipynb		transforms.ipynb