search

dusty-nv / pytorch-segmentation

Training of semantic segmentation networks with PyTorch
34 stars 39 forks source link

PyTorch Version #1

Open wagonhelm opened 4 years ago

wagonhelm commented 4 years ago

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 commented 4 years ago

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.