儲存訓練參數權重(torch.save())
以字典的方式儲存更多訊息
神經網路基礎模型可見此文,本文著重於神經網路訓練的基本元素,包含:model, loss function, optimizer。
Model