adjust tensorflow version

SERVIR · Jun 23, 2024 · 05e2d9a · 05e2d9a
1 parent 62650c3
commit 05e2d9a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/aces/model_trainer.py b/aces/model_trainer.py
@@ -282,7 +282,6 @@ def lr_scheduler(epoch):
 
         self.model_callbacks = model_callbacks
 
-        print(self.model)
         self.history = self.model.fit(
             x=self.TRAINING_DATASET,
             epochs=self.config.EPOCHS,

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 numpy
-tensorflow>=2.9.3
+tensorflow>=2.9.3,<=2.15.0
 earthengine-api
 python-dotenv>=1.0.0
 matplotlib