search

bryandlee / repurpose-gan

Implementation of Repurposing GANs for One-shot Semantic Part Segmentation
126 stars 17 forks source link

Can you save the model somehow? #13

Closed MationPlays closed 2 years ago

MationPlays commented 2 years ago

Hello, can the trained FewShotCNN model saved somehow to compare its performance with other segmentation approaches? the net can be evaluated but can I save it? Thanks for help!

MationPlays commented 2 years ago

this is how you save: torch.save(model.state_dict(), "path/model.pt")