search

adambielski / siamese-triplet

Siamese and triplet networks with online pair/triplet mining in PyTorch
BSD 3-Clause "New" or "Revised" License
3.1k stars 633 forks source link

Save Model Path? #38

Closed dvlshah closed 4 years ago

dvlshah commented 4 years ago

I am training the Triplet MNIST model and couldn't find the path where the model is saved?

adambielski commented 4 years ago

It's not saved in the visualization examples. If you need to save the model just save it as you would normally do in PyTorch, it's all in PyTorch documentation (torch.save(...)).