neural-amp-modeler

commit 1b22d96be8261422064dfa3baf5fdf1849ceb7d7
parent 145ba50fcd15329abfa7d5b28abe0e8e32ccf629
Author: Mike Oliphant <[email protected]>
Date:   Thu, 20 Apr 2023 20:45:39 -0700

Move "metadata" before "weights" in exported model (#193)

* Insert "metadata" before "weights"

* Don't need to add "metadata" here
Diffstat:
M
nam/models/_base.py
 | 
1
-
M
nam/models/_exportable.py
 | 
1
+

2 files changed, 1 insertion(+), 1 deletion(-)
diff --git a/nam/models/_base.py b/nam/models/_base.py
@@ -93,7 +93,6 @@ class _Base(nn.Module, InitializableFromConfig, Exportable):
 
     def _get_export_dict(self):
         d = super()._get_export_dict()
-        d["metadata"] = {}
         return d
 
 
diff --git a/nam/models/_exportable.py b/nam/models/_exportable.py
@@ -103,5 +103,6 @@ class Exportable(abc.ABC):
             "version": __version__,
             "architecture": self.__class__.__name__,
             "config": self._export_config(),
+            "metadata": {},
             "weights": self._export_weights().tolist(),
         }