本文主要讲述:
- self.modue和self.children的区别与联系
说实话,我真的只想讲参数初始化方式,但总感觉在偏离的道路上越走越远。。。
在看一些pytorch文章讲述自定义参数初始化方式时,使用到了self.modules()
和self.children()
函数,觉得还是需要讲解一下的。
不如直接看一下代码:
import torch
from torch import nn
# hyper parameters
in_dim=1
n_hidden_1=1
n_hidden_2=1
out_dim=1
class Net(nn.Module):
def __init__(self, in_dim, n_hidden_1, n_hidden_2, out_dim):
super().__init__()