ruffedisorted

Author: Johanmkr

CollaborativeCoding/models/johan_model.py

@@ -14,15 +14,15 @@ class JohanModel(nn.Module):
         Shape of the input image (C, H, W).
     num_classes : int
         Number of classes in the dataset.
-        
+
     Processing Images
     -----------------
     Input: (N, C, H, W)
         N: Batch size
         C: Number of input channels
         H: Height of the input image
         W: Width of the input image
-        
+
     Example:
     Grayscale images (like MNIST) have C = 1.
     Input shape: (N, 1, 28, 28)

main.py

@@ -1,11 +1,11 @@
 import numpy as np
 import torch as th
 import torch.nn as nn
-import wandb
 from torch.utils.data import DataLoader
 from torchvision import transforms
 from tqdm import tqdm
 
+import wandb
 from CollaborativeCoding import (
     MetricWrapper,
     createfolders,