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:latestTrain from a config:
cd /home/user/atc
python3 ./src/train.py --training-config ./src/training_config.iniRun 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 2048Run 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