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