(self, inputs)
| 99 | self.sum_error = SumErrorLayer() |
| 100 | |
| 101 | def forward(self, inputs): |
| 102 | y = self.linear1(inputs) |
| 103 | y = self.relu(y) |
| 104 | y = self.linear2(y) |
| 105 | return y |
| 106 | |
| 107 | def train_one_batch(self, x, y, dist_option, spars, synflow_flag): |
| 108 | # print ("in train_one_batch") |