File size: 185 Bytes
3f9c425 |
1 2 3 4 5 6 7 8 |
# @package train.optimizer
_target_: torch.distributed.optim.ZeroRedundancyOptimizer
_recursive_: True
optimizer_class:
_target_: torch.optim.__getattribute__
_args_:
- "AdamW"
|