5,323
edits
PyTorch: Difference between revisions
→Building a Model
| Line 73: | Line 73: | ||
* In your class include all other modules you need during init. | * In your class include all other modules you need during init. | ||
** If you have a list of modules, make sure to wrap them in <code>nn.ModuleList</code> or <code>nn.Sequential</code> so they are properly recognized. | ** If you have a list of modules, make sure to wrap them in <code>nn.ModuleList</code> or <code>nn.Sequential</code> so they are properly recognized. | ||
* Wrap any parameters for you model in <code>nn.Parameter(weight, requires_grad=True)</code>. | |||
* Write a forward pass for your model. | * Write a forward pass for your model. | ||