feat: adding model syncing helpers to ivy.stateful.utilites.py

ivy-llc · Sep 18, 2024 · 47ca552 · 47ca552
1 parent f3ad98d
commit 47ca552
Show file tree

Hide file tree

Showing 2 changed files with 743 additions and 0 deletions.
diff --git a/ivy/stateful/__init__.py b/ivy/stateful/__init__.py
@@ -16,3 +16,5 @@
 from .optimizers import *
 from . import sequential
 from .sequential import *
+from . import utilities
+from .utilities import sync_models_torch