(self)
| 255 | state_dict_key = "amp_scaler" |
| 256 | |
| 257 | def __init__(self): |
| 258 | self._scaler = torch.cuda.amp.GradScaler() |
| 259 | |
| 260 | def __call__(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True): |
| 261 | self._scaler.scale(loss).backward(create_graph=create_graph) |
nothing calls this directly
no outgoing calls
no test coverage detected