:param optimizer: String (name of optimizer) or optimizer instance. See [optimizers](https://pytorch.org/docs/stable/optim.html). :param loss: String (name of objective function) or objective function. See [losses](https://pytorch.org/docs/stable/nn.functional.html#loss-functions).
(self, optimizer,
loss=None,
metrics=None,
)
| 437 | self.aux_loss = aux_loss * alpha |
| 438 | |
| 439 | def compile(self, optimizer, |
| 440 | loss=None, |
| 441 | metrics=None, |
| 442 | ): |
| 443 | """ |
| 444 | :param optimizer: String (name of optimizer) or optimizer instance. See [optimizers](https://pytorch.org/docs/stable/optim.html). |
| 445 | :param loss: String (name of objective function) or objective function. See [losses](https://pytorch.org/docs/stable/nn.functional.html#loss-functions). |
| 446 | :param metrics: List of metrics to be evaluated by the model during training and testing. Typically you will use `metrics=['accuracy']`. |
| 447 | """ |
| 448 | self.metrics_names = ["loss"] |
| 449 | self.optim = self._get_optim(optimizer) |
| 450 | self.loss_func = self._get_loss_func(loss) |
| 451 | self.metrics = self._get_metrics(metrics) |
| 452 | |
| 453 | def _get_optim(self, optimizer): |
| 454 | if isinstance(optimizer, str): |