Skip to content

Latest commit

 

History

History
44 lines (36 loc) · 960 Bytes

File metadata and controls

44 lines (36 loc) · 960 Bytes

Run Log

Start a container with project output folders mounted:

docker run -it --rm --shm-size 32G --gpus all \
  -v ./atc/save:/home/user/atc/save \
  -v ./atc/opensky_save:/home/user/atc/opensky_save \
  -v ./atc/extra_save:/home/user/atc/extra_save \
  atc-sim:latest

Train from a config:

cd /home/user/atc
python3 ./src/train.py --training-config ./src/training_config.ini

Run inference from a checkpoint:

python3 ./src/inference.py \
  --save-folder ./save/<run-name>-inf \
  --ckpt-path ./save/<run-name>/ckpt/<step>.pt \
  --device cuda:0 \
  --batch-size 2048

Run simulator interaction experiments:

python3 ./src/interaction.py \
  --num-proc 10 \
  --num-exp 2 \
  --save-folder ./extra_save/<run-name>-int \
  --ckpt-folder ./save/<run-name>/ckpt \
  --ckpt-idx "0:100000:2500" \
  --num-timestamps 5000 \
  --nargs-try-ils 0 \
  --nargs-take-idx 6 8 10 12 \
  --device cuda:0 \
  --save-plot