SGM_pytorch

In process of converting SGM from Tensorflow to Pytorch

Dependencies

Minimum python3.6 required

pip install torch==1.4.0 torchvision==0.5.0 -f https://download.pytorch.org/whl/cu100/torch_stable.html -> using CUDA 10.0

pip install numpy

pip install kornia==0.2.2 -> computer vision library for pytorch