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 |