a | b/run.sh | ||
---|---|---|---|
1 | #!/usr/bin/env bash |
||
2 | python -m nips.round2_train \ |
||
3 | --num-timesteps=1000000 \ |
||
4 | --num-steps=256 \ |
||
5 | --num-minibatches=16 \ |
||
6 | --num-cpus=24 --num-casks=8 \ |
||
7 | --num-gpus=1 \ |
||
8 | --save-interval=1 \ |
||
9 | --seed=30 --repeat=2 \ |
||
10 | --checkpoint-path=checkpoints/00000 |