PyTorch: TORCH.NN

Pytorch提供了torch.nn、torch.optim、Dataset和DataLoader這些設計優雅的模塊和類以幫助使用者創建和訓練神經網絡。 爲了最大化利用這些模塊和類的功能,並使用它們做出適用於你所研究問題的模型,你需要真正理解他們是如何工作的。 爲了做到這一點,我們首先基於MNIST數據集訓練一個沒有任何特徵的簡單神經網絡。 最開始我們只會用到PyTorch中最基本的tensor功
相關文章
相關標籤/搜索