refactoring

CardioKit · Jan 22, 2024 · 832d4fd · 832d4fd
1 parent 1067631
commit 832d4fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -48,7 +48,7 @@ def main(parameters):
         #ModelCheckpoint(filepath=base_path + 'model/', monitor='loss', save_best_only=True, verbose=0),
         ReconstructionPlot(train, parameters['index_tracked_sample'], base_path + 'training/reconstruction/',
                            period=parameters['period_reconstruction_plot']),
-        CollapseCallback(val),
+        #CollapseCallback(val),
         EarlyStopping(monitor="val_loss", patience=parameters['early_stopping'])
     ]