Merge pull request #9 from CapitalRobotics/adaptive

update
CapitalRobotics · Nov 17, 2024 · 5c80cb8 · 5c80cb8
2 parents b11b346 + 8e4b9e0
commit 5c80cb8
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/atem_core/__init__.py b/atem_core/__init__.py
@@ -0,0 +1,2 @@
+from .adaptive import *
+from .core import *
diff --git a/atem_core/adaptive/__init__.py b/atem_core/adaptive/__init__.py
@@ -0,0 +1,6 @@
+from .test_interp import interpret_model
+
+__all__ = [
+    "interpret_model",
+
+]
diff --git a/atem_core/adaptive/train.py b/atem_core/adaptive/train.py
@@ -26,4 +26,9 @@
 converter = tf.lite.TFLiteConverter.from_keras_model(model)
 tflite_model = converter.convert()
 with open("adaptive_model.tflite", "wb") as f:
-    f.write(tflite_model)
+    f.write(tflite_model)
+
+
+
+
+
diff --git a/atem_core/core/__init__.py b/atem_core/core/__init__.py
@@ -0,0 +1,2 @@
+from .task_gen import generate_task_data
+from .interp_test import *
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		from .task_gen import generate_task_data
		from .interp_test import *