Uni-Core icon indicating copy to clipboard operation
Uni-Core copied to clipboard

Published 20 hours ago • verified publisher icon dptech-corp

replace the EMA implementation with pytorch-native AveragedModel

Open leasunhy opened this issue 7 months ago • 0 comments

This PR replaces the custom EMA implementation with the one in PyTorch.

Note that this PR breaks backward compatibility: it cannot load old-format checkpoints that were generated with ema enabled.

leasunhy avatar Jul 18 '24 10:07 leasunhy