r
(self)
| 1654 | self.module.train(mode) |
| 1655 | |
| 1656 | def eval(self): |
| 1657 | r"""""" |
| 1658 | |
| 1659 | self.warn_unscaled_loss = True |
| 1660 | self.module.train(False) |
| 1661 | |
| 1662 | def _scale_loss_by_gas(self, prescaled_loss): |
| 1663 | if isinstance(prescaled_loss, torch.Tensor): |
no test coverage detected