mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
2,347 Commits 141 Branches 65 Tags
Commit Graph

9 Commits

Author SHA1 Message Date
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