Skip to content

Latest commit

 

History

History
34 lines (23 loc) · 553 Bytes

README.md

File metadata and controls

34 lines (23 loc) · 553 Bytes
Raw

CIFAR10 TorchPie

Requirements

  • Pytorch
  • TorchPie
  • TensorBoard

Install TorchPie

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

Usage

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.

Development

ln -s ../torchpie/torchpie ./