add API demo from #1
This commit is contained in:
1 parent
06e1b120e1
commit
3d8d5ee5d5
5 files changed
+191
-178
No files matched your search
+1
-1
@@ -142,7 +142,7 @@ def load_valuehead_params(model: torch.nn.Module, checkpoint_dir: os.PathLike) -
|
||||
|
||||
|
||||
def smooth(scalars: List[float], weight: Optional[float] = 0.9) -> List[float]:
|
||||
"""
|
||||
r"""
|
||||
EMA implementation according to TensorBoard.
|
||||
"""
|
||||
last = scalars[0]
|
||||
|
||||
Reference in new issue
Block a user