feat: optuna optimization performed

2025-11-27 22:35:27 +01:00 · 2025-11-27 22:35:27 +01:00 · fe207962de
commit fe207962de
parent 2ab4abdf93
5 changed files with 15 additions and 18 deletions
--- a/CNN-model/trainers/train.py
+++ b/CNN-model/trainers/train.py
@ -56,6 +56,4 @@ def train(
            avg_loss = sum(losses) / len(losses)
            avg_validation_losses.append(avg_loss)

-        tqdm.write(f"epoch: {epoch + 1}, avg val loss = {avg_loss:.4f}")
-
    return avg_training_losses, avg_validation_losses