(self, loss: th.Tensor)
| 174 | zero_grad(self.model_params) |
| 175 | |
| 176 | def backward(self, loss: th.Tensor): |
| 177 | if self.use_fp16: |
| 178 | loss_scale = 2 ** self.lg_loss_scale |
| 179 | (loss * loss_scale).backward() |
| 180 | else: |
| 181 | loss.backward() |
| 182 | |
| 183 | def optimize(self, opt: th.optim.Optimizer): |
| 184 | if self.use_fp16: |
nothing calls this directly
no outgoing calls
no test coverage detected