pytorch-segmentation icon indicating copy to clipboard operation
pytorch-segmentation copied to clipboard

Published 20 hours ago • verified publisher icon dusty-nv

PyTorch Version

Open wagonhelm opened this issue 5 years ago • 1 comments

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

wagonhelm avatar Dec 31 '19 01:12 wagonhelm

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.

bhusalsantosh avatar Jan 17 '20 21:01 bhusalsantosh