(self, *input, **kwargs)
| 236 | return self.__class__.__name__ |
| 237 | |
| 238 | def __call__(self, *input, **kwargs): |
| 239 | if self.training: |
| 240 | return self.train_one_batch(*input, **kwargs) |
| 241 | else: |
| 242 | return self.forward(*input, **kwargs) |
| 243 | |
| 244 | def save_states(self, fpath, aux_states={}): |
| 245 | """Save states. |
nothing calls this directly
no test coverage detected