Merge pull request #16 from IvanKuchin/development

disable terminateonNaN
IvanKuchin · Jul 25, 2024 · 9d06683 · 9d06683
2 parents 2362b38 + c818238
commit 9d06683
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train_segmentation.py b/train_segmentation.py
@@ -103,7 +103,8 @@ def main():
  reduce_lr_on_plateau,
  csv_logger,
  # early_stopping,
- tf.keras.callbacks.TerminateOnNaN()],
+ # tf.keras.callbacks.TerminateOnNaN()
+ ],
  verbose = 1,
  # workers = 2
  )