[英]Custom NN architecture using Pytorch
我正在嘗試使用Pytorch創建自定義的CNN架構。 我希望擁有與僅使用numpy進行架構時得到的控件相同的控件。
我是Pytorch的新手,如果可能的話,我希望看到一些沒有nn.module類而實現的CNN代碼示例。
您必須在自定義類中實現backward()
函數。
但是從您的問題尚不清楚
nn.Sequential(nn.Conv2d( ...)
)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.