| 141 | self.assertEqual(len(out), 2) |
| 142 | |
| 143 | def test_layerlist(self): |
| 144 | print('-' * 20, 'layerlist', '-' * 20) |
| 145 | |
| 146 | class MyModel(Model): |
| 147 | |
| 148 | def __init__(self): |
| 149 | super(MyModel, self).__init__() |
| 150 | self.layers = LayerList([Dense(50, in_channels=100), Dropout(0.9), Dense(10, in_channels=50)]) |
| 151 | |
| 152 | def forward(self, x): |
| 153 | return self.layers(x) |
| 154 | |
| 155 | net = MyModel() |
| 156 | self.assertEqual(net._nodes_fixed, False) |
| 157 | |
| 158 | data = np.random.normal(size=[4, 100]).astype(np.float32) |
| 159 | out = net(data, is_train=False) |
| 160 | |
| 161 | self.assertEqual(net._nodes_fixed, True) |
| 162 | self.assertEqual(net.layers._nodes_fixed, True) |
| 163 | self.assertEqual(net.layers[0]._nodes_fixed, True) |
| 164 | self.assertEqual(net.layers[1]._nodes_fixed, True) |
| 165 | self.assertEqual(net.layers[2]._nodes_fixed, True) |
| 166 | |
| 167 | def test_ModelLayer(self): |
| 168 | print('-' * 20, 'ModelLayer', '-' * 20) |