pytorch網絡搭建

網絡搭建技巧 善用Sequential 善用Sequential 首先將torch.nn構建的網絡層轉化爲list類型,加[ ] [conv , relu , norm] 將不同網絡層list結合,相加+ list類型通過 * 操作可以提取列表 *self.model = nn.Sequential(model)
相關文章
相關標籤/搜索