Load running mean and var from weight file

2025-11-06 09:44:28 +00:00 · 2024-08-25 19:33:33 +02:00
parent 9704d0d53e
commit bc9bff10cd
4 changed files with 56 additions and 3 deletions
--- a/include/model/model.hpp
+++ b/include/model/model.hpp
@@ -15,6 +15,8 @@ namespace CUDANet {
 enum TensorType {
    WEIGHT,
    BIAS,
+    RUNNING_MEAN,
+    RUNNING_VAR
 };

 struct TensorInfo {