Defines the computation performed in every forward propagation. Should be overridden by all subclasses. Args: *input: the input training data for the model Returns: out: the outputs of the forward propagation.
(self, *input)
| 187 | self.sequential = sequential |
| 188 | |
| 189 | def forward(self, *input): |
| 190 | """Defines the computation performed in every forward propagation. |
| 191 | |
| 192 | Should be overridden by all subclasses. |
| 193 | |
| 194 | Args: |
| 195 | *input: the input training data for the model |
| 196 | |
| 197 | Returns: |
| 198 | out: the outputs of the forward propagation. |
| 199 | """ |
| 200 | raise NotImplementedError |
| 201 | |
| 202 | def train_one_batch(self, *input, **kwargs): |
| 203 | """Defines the computation performed in every training iteration |