nn.Module屬性

nn.Module屬性 所有的網絡層和所有的網絡層都是繼承nn.Module,對他的瞭解十分有必要。 最常用的就是Parameters和Module的字典 在對者兩個賦值時,會被setAttr函數攔截進行判斷,被賦值函數是Parameters類型還是Module類型,然後再進行相應操作
相關文章
相關標籤/搜索