(self, *input, **kwargs)
| 259 | return self.__class__.__name__ |
| 260 | |
| 261 | def __call__(self, *input, **kwargs): |
| 262 | # print ("in pkg model.py __call__") |
| 263 | if self.training: |
| 264 | # print ("in pkg model.py train_one_batch") |
| 265 | # print ("self: ", self) |
| 266 | # print ("self.num_classes: ", self.num_classes) |
| 267 | # print ("input[0].shape: ", input[0].shape) |
| 268 | return self.train_one_batch(*input, **kwargs) |
| 269 | else: |
| 270 | # print ("in pkg model.py forward") |
| 271 | return self.forward(*input, **kwargs) |
| 272 | |
| 273 | def save_states(self, fpath, aux_states={}): |
| 274 | """Save states. |
nothing calls this directly
no test coverage detected