How can i use the pretrained model

nv-tlabs / GSCNN

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

https://nv-tlabs.github.io/GSCNN/

Other

916 stars 200 forks source link

How can i use the pretrained model #18

Closed XuShoweR closed 4 years ago

XuShoweR commented 5 years ago

i want to train another dataset, how can i use the pretrained model is this correct? python train.py --snapshot checkpoints/best_cityscapes_checkpoint.pth

tovacinni commented 4 years ago

Yes, that's correct! Make sure you implement the dataloader etc such that you can train though.