(self)
| 26 | state_dict_key = "amp_scaler" |
| 27 | |
| 28 | def __init__(self): |
| 29 | self._scaler = torch.cuda.amp.GradScaler() |
| 30 | |
| 31 | def __call__(self, loss, optimizer, clip_grad=None, parameters=None, create_graph=False, update_grad=True,retain_graph=False): |
| 32 | self._scaler.scale(loss).backward(create_graph=create_graph, retain_graph=retain_graph) |
nothing calls this directly
no outgoing calls
no test coverage detected