diff --git a/.gitignore b/.gitignore
index 49835a7..e3356ea 100644
--- a/.gitignore
+++ b/.gitignore
@@ -12,3 +12,5 @@ output.html
 *.sw?
 TAGS
 uv.lock
+/ignore
+*.log
diff --git a/train/requirements.txt b/train/requirements.txt
index 1734b67..dbc4f76 100644
--- a/train/requirements.txt
+++ b/train/requirements.txt
@@ -6,6 +6,8 @@ imutils
 scipy
 tensorflow-addons # for connected_components, depublished and only compatible with tensorflow < 2.16
 tensorflow < 2.16 # for tensorflow-addons, so only needed in training
-tf-keras # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)
+tf-keras < 2.16 # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)
 tf-data < 2.16 # for tensorflow-addons, so only needed in training
 protobuf < 5 # for tensorflow-addons, so only needed in training
+tf2onnx
+ml_dtypes >= 0.5