Refactor/optim (#272)

tracel-ai · Apr 5, 2023 · ca8ee07 · ca8ee07
1 parent 0f6172f
commit ca8ee07
Show file tree

Hide file tree

Showing 30 changed files with 744 additions and 652 deletions.
diff --git a/burn-core/src/module/mod.rs b/burn-core/src/module/mod.rs
@@ -1,7 +1,5 @@
 mod base;
 mod param;
-mod state;
 
 pub use base::*;
 pub use param::*;
-pub use state::*;
diff --git a/burn-core/src/module/param/id.rs b/burn-core/src/module/param/id.rs
@@ -17,6 +17,12 @@ impl From<&str> for ParamId {
     }
 }
 
+impl From<String> for ParamId {
+    fn from(value: String) -> Self {
+        Self { value }
+    }
+}
+
 impl Default for ParamId {
     fn default() -> Self {
         Self::new()

diff --git a/burn-core/src/module/param/primitive.rs b/burn-core/src/module/param/primitive.rs
@@ -145,6 +145,7 @@ where
 impl<const N: usize, T, B> Module<B> for [T; N]
 where
     T: Module<B> + Debug + Send + Sync + Clone + Copy,
+    T::Record: Debug,
     B: Backend,
 {
     type Record = [T::Record; N];
@@ -201,7 +202,9 @@ where
 impl<const N: usize, T, B> ADModule<B> for [T; N]
 where
     T: ADModule<B> + Debug + Send + Sync + Clone + Copy,
-    T::InnerModule: Copy,
+    T::InnerModule: Copy + Debug,
+    <T::InnerModule as Module<B::InnerBackend>>::Record: Debug,
+    <T as Module<B>>::Record: Debug,
     B: ADBackend,
 {
     type InnerModule = [T::InnerModule; N];

diff --git a/burn-core/src/module/state.rs b/burn-core/src/module/state.rs