Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard
verified publisher icon milesial

After training the model, how to save it into model.pth

Open dugurrn opened this issue 3 years ago • 1 comments

Sorry I know this might be a really stupid question. Should I just use any of the checkpoint_epoch.pth? Or where can I find the model.pth and save it?

dugurrn avatar Dec 12 '22 08:12 dugurrn

The checkpoints contain the model weights yes. You should use the checkpoint for the epoch you prefer (better accuracy), usually the latest.

milesial avatar Dec 12 '22 10:12 milesial