diff --git a/src/models/sofia_modelle/CNN.py b/src/models/sofia_modelle/CNN.py index d63be5b028e83ff22d9b31f3860a9891290304cf..6724bc9940f091469090c825a71238f7aaba0a02 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