From 2f13bb15c6c521b2af723a07d8bbad51873ec524 Mon Sep 17 00:00:00 2001
From: DesireeWyrzylala <74858504+DesireeWyrzylala@users.noreply.github.com>
Date: Thu, 6 Feb 2025 10:38:49 +0100
Subject: [PATCH] fix cnn
---
src/models/sofia_modelle/CNN.py | 32 ++++++++++++++++----------------
1 file changed, 16 insertions(+), 16 deletions(-)
diff --git a/src/models/sofia_modelle/CNN.py b/src/models/sofia_modelle/CNN.py
index d63be5b..6724bc9 100644
--- a/src/models/sofia_modelle/CNN.py
+++ b/src/models/sofia_modelle/CNN.py
@@ -278,26 +278,26 @@ class CNN():
f.write(str(model_stats))
-#################################### TEST UND TRAINING #################################
-# Training
-num_epochs = 20
+# #################################### TEST UND TRAINING #################################
+# # Training
+# num_epochs = 20
-for epoch in range(num_epochs):
- model.train()
- running_loss = 0.0
+# for epoch in range(num_epochs):
+# model.train()
+# running_loss = 0.0
- for inputs, targets in dataloader:
- inputs, targets = inputs.to(device), targets.to(device)
+# for inputs, targets in dataloader:
+# inputs, targets = inputs.to(device), targets.to(device)
- optimizer.zero_grad()
- outputs = model(inputs)
- loss = criterion(outputs, targets)
+# optimizer.zero_grad()
+# outputs = model(inputs)
+# loss = criterion(outputs, targets)
- loss.backward()
- optimizer.step()
+# loss.backward()
+# optimizer.step()
- running_loss += loss.item()
+# running_loss += loss.item()
- print(f"Epoch {epoch+1}, Loss: {running_loss/len(dataloader)}")
+# print(f"Epoch {epoch+1}, Loss: {running_loss/len(dataloader)}")
-# Test
\ No newline at end of file
+# # Test
\ No newline at end of file
--
GitLab