Training Process - Geomni/high-res-stereo GitHub Wiki

Docker Run On Omen

docker run -it --gpus all -v ~/high-res-stereo/model:/workspace/model -v ~/high-res-stereo/data:/workspace/data -v ~/.aws:/root/.aws --shm-size="8G" <docker image>

Docker run EC2

docker run -it --gpus all -v /data/model:/workspace/model -v /data/data:/workspace/data --shm-size="8G"

Run training script p3.2xlarge

python train.py --savemodel /workspace/model --database /workspace/data --persist_to_s3 --experiment_name test_p3 --use_tiny_dataset  --epochs 10 --batchsize 10

Setup Instance

https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#installing-on-amazon-linux