|
|
|
@@ -259,8 +259,13 @@ class CellList(_CellListBase, Cell): |
|
|
|
>>> cell_ls.append(relu) |
|
|
|
>>> cell_ls |
|
|
|
CellList< |
|
|
|
(0): Conv2d<input_channels=100, ..., bias_init=None> |
|
|
|
(1): BatchNorm2d<num_features=20, ..., moving_variance=Parameter (name=variance)> |
|
|
|
(0): Conv2d<input_channels=100, output_channels=20, kernel_size=(3, 3),stride=(1, 1), pad_mode=same, |
|
|
|
padding=0, dilation=(1, 1), group=1, has_bias=Falseweight_init=normal, bias_init=zeros, format=NCHW> |
|
|
|
(1): BatchNorm2d<num_features=20, eps=1e-05, momentum=0.09999999999999998, gamma=Parameter (name=1.gamma, |
|
|
|
shape=(20,), dtype=Float32, requires_grad=True), beta=Parameter (name=1.beta, shape=(20,), dtype=Float32, |
|
|
|
requires_grad=True), moving_mean=Parameter (name=1.moving_mean, shape=(20,), dtype=Float32, |
|
|
|
requires_grad=False), moving_variance=Parameter (name=1.moving_variance, shape=(20,), dtype=Float32, |
|
|
|
requires_grad=False)> |
|
|
|
(2): ReLU<> |
|
|
|
> |
|
|
|
""" |
|
|
|
|