Ross Wightman
|
24f6d4f7f8
|
Fix #2127 move to ema device
|
2024-04-10 21:29:09 -07:00 |
|
Ross Wightman
|
bee0471f91
|
forward() pass through for ema model, flag for ema warmup, comment about warmup
|
2024-02-03 16:24:45 -08:00 |
|
Ross Wightman
|
dd84ef2cd5
|
ModelEmaV3 and MESA experiments
|
2024-02-02 09:45:04 -08:00 |
Jerome Rony
|
3491506fec
|
Add foreach option for faster EMA
|
2022-11-30 14:06:58 -05:00 |
|
Jerome Rony
|
6ec5cd6a99
|
Use in-place operations for EMA
|
2022-11-17 11:53:29 -05:00 |
|
Ross Wightman
|
4ca52d73d8
|
Add separate set and update method to ModelEmaV2
|
2020-12-03 10:05:09 -08:00 |
|
Ross Wightman
|
27bbc70d71
|
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.
|
2020-11-29 16:22:19 -08:00 |
|
Ross Wightman
|
9214ca0716
|
Simplifying EMA...
|
2020-11-16 12:51:52 -08:00 |
|
Ross Wightman
|
532e3b417d
|
Reorg of utils into separate modules
|
2020-09-07 13:58:09 -07:00 |