(self, inputs)
| 95 | self.sum_error = SumErrorLayer() |
| 96 | |
| 97 | def forward(self, inputs): |
| 98 | y = self.linear1(inputs) |
| 99 | y = self.relu(y) |
| 100 | y = self.linear2(y) |
| 101 | return y |
| 102 | |
| 103 | def train_one_batch(self, x, y, synflow_flag, dist_option, spars): |
| 104 | out = self.forward(x) |