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