isorted

Author: hzavadil98

main.py

@@ -3,11 +3,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 utils import MetricWrapper, createfolders, get_args, load_data, load_model
 
 

utils/load_model.py

@@ -1,6 +1,6 @@
 import torch.nn as nn
 
-from .models import ChristianModel, JanModel, MagnusModel, SolveigModel, JohanModel
+from .models import ChristianModel, JanModel, JohanModel, MagnusModel, SolveigModel
 
 
 def load_model(modelname: str, *args, **kwargs) -> nn.Module:

utils/models/__init__.py

@@ -2,6 +2,6 @@
 
 from .christian_model import ChristianModel
 from .jan_model import JanModel
+from .johan_model import JohanModel
 from .magnus_model import MagnusModel
 from .solveig_model import SolveigModel
-from .johan_model import JohanModel