(self, args)
| 311 | state_dict_key = "amp_scaler" |
| 312 | |
| 313 | def __init__(self, args): |
| 314 | self._scaler = ShardedGradScaler(enabled=args.precision in ["fp16"]) |
| 315 | |
| 316 | def __call__(self, loss, optimizer, model, clip_grad=None, parameters=None, create_graph=False, update_grad=True): |
| 317 | if update_grad: |
nothing calls this directly
no outgoing calls
no test coverage detected