- Pytorch
- TorchPie
- TensorBoard
Install TorchPie
pip install git+https://git.dev.tencent.com/SunDoge/torchpie.gitUse seleted GPUs
python -m torchpie.distributed.launch --gpus 0,1 main.py -c config/default.conf -e exps/resnet20-2gpuUse all GPUs
python -m torchpie.distributed.launch main.py -c config/default.conf -e exps/resnet20-2gpuMake sure you use different experiment-path every time you run.
ln -s ../torchpie/torchpie ./