Olamide Isreal
a3ffec6a07
Some checks failed
Code Quality Main / code-quality (push) Has been cancelled
Release Drafter / update_release_draft (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.10) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.8) (push) Has been cancelled
Tests / run_tests_ubuntu (ubuntu-latest, 3.9) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.10) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.8) (push) Has been cancelled
Tests / run_tests_macos (macos-latest, 3.9) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.10) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.8) (push) Has been cancelled
Tests / run_tests_windows (windows-latest, 3.9) (push) Has been cancelled
Tests / code-coverage (push) Has been cancelled
11 lines
397 B
YAML
11 lines
397 B
YAML
# https://github.com/NVIDIA/NeMo/blob/main/nemo/collections/common/callbacks/ema.py
|
|
|
|
# Maintains an exponential moving average (EMA) of model weights.
|
|
# Look at the above link for more detailed information regarding the original implementation.
|
|
ema:
|
|
_target_: flowdock.models.components.callbacks.ema.EMA
|
|
decay: 0.999
|
|
validate_original_weights: false
|
|
every_n_steps: 4
|
|
cpu_offload: false
|