Merge 503946e into e16cebc

fuzzylabs · Feb 10, 2025 · b10bd8f · b10bd8f
2 parents e16cebc + 503946e
commit b10bd8f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/model_training/data_preparation/measurement_utils.py b/model_training/data_preparation/measurement_utils.py
@@ -12,7 +12,7 @@
     "layer_run_time": "average_run_time",
 }
 
-LAYER_TYPES = ["CaskPooling", "CaskConvolution", "CaskGemmConvolution"]
+LAYER_TYPES = ["Conv2d", "AvgPool2d","MaxPool2d", "AdaptiveAvgPool2d", "AdaptiveMaxPool2d", "Linear"]
 
 
 def preprocess_measurement_data(

diff --git a/model_training/training_data.dvc b/model_training/training_data.dvc
@@ -1,6 +1,6 @@
 outs:
-- md5: afe70c9bc655ffa0fc2e631b83088972.dir
-  size: 312440
-  nfiles: 55
+- md5: d79906bbdced4b3029a4764780c0973c.dir
+  size: 50216
+  nfiles: 34
   hash: md5
   path: training_data