working server

2026-04-28 23:33:00 +02:00
parent a271c738d7
commit 1e7d9686d4
9 changed files with 90 additions and 6 deletions
@@ -16,13 +16,14 @@ class Bird_CNN(nn.Module):

            nn.Conv2d(c_hidden, c_hidden, kernel_size, padding=kernel_size//2),
            nn.ReLU(),
+
            nn.Flatten(),
            nn.Linear(c_hidden * img_height * img_width, c_out)
        )

    def forward(self, x):
        return self.model(x)
-    
+

 def trainCNN(model, optimizer, loss_module, train_data_loader, validation_data_loader, device, num_epochs, SAVE_PATH, save=False):