8 lines
185 B
YAML
8 lines
185 B
YAML
# @package train.optimizer
|
|
_target_: torch.distributed.optim.ZeroRedundancyOptimizer
|
|
_recursive_: True
|
|
optimizer_class:
|
|
_target_: torch.optim.__getattribute__
|
|
_args_:
|
|
- "AdamW"
|