File size: 196 Bytes
3f9c425 |
1 2 3 4 5 6 7 8 |
# @package train.optimizer
_target_: torch.distributed.optim.ZeroRedundancyOptimizer
_recursive_: True
optimizer_class:
_target_: apex.optimizers.FusedAdam
_partial_: True
adam_w_mode: True
|