【深度學習】Pytorch學習筆記(一)

Pytorch中需要自己定義網絡模型,該模型需要封裝到一個自定義的類中,該類只是一個子類,其繼承的父類爲torch.nn.Module,可見如下樹形結構圖: Module實際又是繼承了Object類,關於爲什麼要繼承Object類有興趣的可以看這篇博客MRO算法 也就是說,自定義的模型除了要有__init__()構造函數之外,還必須有forward函數,重寫父類Module的該方法。 在子類的__
相關文章
相關標籤/搜索