about save .pth in training

nv-tlabs / GSCNN

Gated-Shape CNN for Semantic Segmentation (ICCV 2019)

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

Other

915 stars 200 forks source link

Open JessicaLi0613 opened 4 years ago

JessicaLi0613 commented 4 years ago

hello blogger! It seems that there is no code about saving .pth weights in train.py? Or other files exist the saving code?

ayinaaaaaa commented 4 years ago

hello blogger! It seems that there is no code about saving .pth weights in train.py? Or other files exist the saving code?

hi,the code about saving .pth weights seems to be in until/misc.py.