Skip to content

Commit b58595c

Browse files
jemrobinsonjemrobinson
committed
🔧 Add the EMAWeightAveragingCallback to our set of default callbacks
1 parent de4114a commit b58595c

2 files changed

Lines changed: 5 additions & 0 deletions

File tree

‎ice_station_zebra/config/train/callbacks/ema_weight_averaging.yaml‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
ema_weight_averaging:
2+
_target_: ice_station_zebra.callbacks.EMAWeightAveragingCallback
3+
every_n_epochs: 1
4+
every_n_steps: 100

‎ice_station_zebra/config/train/default.yaml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
defaults:
22
- callbacks:
33
- device_stats
4+
- ema_weight_averaging
45
- unconditional_checkpoint
56
- optimizer: default
67
- trainer: default

0 commit comments

Comments
 (0)