diff --git a/README.md b/README.md index 24d910d..e45c2f4 100644 --- a/README.md +++ b/README.md @@ -14,7 +14,8 @@ You can demo a trained model on pairs of images. To predict stereo for Middlebur ``` python demo.py --restore_ckpt ./pretrained_models/sceneflow/sceneflow.pth ``` - + + ## Environment * NVIDIA RTX 3090