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