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"
|