PyTorch Version

[wagonhelm]

Hi There,

I'm having some issues trying to use this to train a custom image segmentation model and having some issues. I see the custom fork for torchvision v.0.3.0 but just wondering which version of torch this was written for.

Thanks

[bhusalsantosh]

I installed torch and torchvision from (https://devtalk.nvidia.com/default/topic/1049071/pytorch-for-jetson-nano-version-1-3-0-now-available/). You might want to see the right working version from (https://devtalk.nvidia.com/default/topic/1069288/jetson-nano/pytorch-amp-torchvision-installation-issue/). I did not install pytorch while building this repo. The segmentation model was working while testing. I am working towards training on custom dataset.