I am copying the example from here:
https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html 78
the function was supposed to return the model, but I had somehow removed that line. Not that it is back, torch.save is working fine! Thank you for making me look there!