Fixed some things. Dry-run OK. Need changes in metric loader

Author: Seilmast

main.py

@@ -113,13 +113,13 @@ def main():
 
     # wandb.login(key=WANDB_API)
     wandb.init(
-        entity="ColabCode-org",
-        # entity="FYS-8805 Exam",
-        project="Test",
+        entity="ColabCode",
+        project="Magnus Runs",
         tags=[args.modelname, args.dataset],
+        config=args,
     )
     wandb.watch(model)
-    exit()
+
     for epoch in range(args.epoch):
         # Training loop start
         trainingloss = []

utils/metrics/EntropyPred.py

@@ -3,7 +3,7 @@
 
 
 class EntropyPrediction(nn.Module):
-    def __init__(self, averages: str = "average"):
+    def __init__(self, averages: str = "mean"):
         """
         Initializes the EntropyPrediction module, which calculates the Shannon Entropy
         of predicted logits and aggregates the results based on the specified method.

utils/models/magnus_model.py

@@ -2,7 +2,7 @@
 
 
 class MagnusModel(nn.Module):
-    def __init__(self, image_shape, num_classes: int, nr_channels: int):
+    def __init__(self, image_shape, num_classes: int, nr_channels: int = 1):
         """
         Initializes the MagnusModel, a neural network designed for image classification tasks.
         The model consists of three linear layers, each with 133 neurons, and uses ReLU activation