Fix missing values as NaN

Esteban Gómez · Esteban Gómez · commit 831c5cb2f132 · 2024-09-26T01:15:45.000+02:00
diff --git a/src/moduleprofiler/profiler.py b/src/moduleprofiler/profiler.py
@@ -380,7 +380,11 @@ def count_params(
             data[n] = {
                 "type": m.__class__.__name__,
                 "trainable_params": 0,
-                "nontrainable_params": 0
+                "trainable_params_dtype": None,
+                "trainable_params_size_bits": 0,
+                "nontrainable_params": 0,
+                "nontrainable_params_dtype": None,
+                "nontrainable_params_size_bits": 0
             }
 
             for p in m.parameters():
