Added a simple test to make sure github actions work

Author: Seilmast

.gitignore

@@ -1,4 +1,5 @@
 Data/*
 Results/*
 Experiments/*
-env/*
+env/*
+test.ipynb

main.py

@@ -2,7 +2,7 @@
 import torch.nn as nn 
 
 import argparse
-from utils import load_metric
+from utils import load_metric, load_model, createfolders
 
 
 
@@ -16,17 +16,50 @@
 
 
 def main():
+    '''
     
+    Parameters
+    ----------
+    
+    Returns
+    -------
+    
+    Raises
+    ------
+    
+    '''
     parser = argparse.ArgumentParser(
         prog='',
         description='',
         epilog='',
     )
+    #Structuture related values
+    parser.add_argument('--datafolder', type=str, default='Data/', help='Path to where data will be saved during training.')
+    parser.add_argument('--resultfolder', type=str, default='Results/', help='Path to where results will be saved during evaluation.')
+    parser.add_argument('--modelfolder', type=str, default='Experiments/', help='Path to where model weights will be saved at the end of training.')
+    parser.add_argument('--savemodel', type=bool, default=False, help='Whether model should be saved or not.')
+    parser.add_argument('--download-data', type=bool, default=False, help='Whether the data should be downloaded or not. Might cause code to start a bit slowly.')
 
-    parser.add_argument('--epoch', type=int, default=20, help='Amount of training epochs the model will do')
+    #Data/Model specific values
+    parser.add_argument('--modelname', type=str, default='MagnusModel', 
+                        choices = ['MagnusModel'], help="Model which to be trained on")
 
+    #Training specific values
+    parser.add_argument('--epoch', type=int, default=20, help='Amount of training epochs the model will do.')
+    parser.add_argument('--learning_rate', type=float, default=0.001, help='Learning rate parameter for model training.')
     args = parser.parse_args()
 
+    createfolders(args)
+    
+    device = 'cuda' if th.cuda.is_available() else 'cpu'
+    
+    #load model
+    model = load_model()
+    model.to(device)
+    
+    
+    criterion = nn.CrossEntropyLoss()
+    optimizer = th.optim.Adam(model.parameters(), lr = args.learning_rate)
 
 
 
@@ -36,7 +69,7 @@ def main():
 
         #Eval loop start
 
-
+        pass 
 
 
 if __name__ == '__main__':

utils/__init__.py

@@ -1 +1,4 @@
-from .load_metric import load_metric
+from .load_metric import load_metric
+from .load_model import load_model
+from .load_data import load_data
+from .createfolders import createfolders

utils/createfolders.py

@@ -0,0 +1,41 @@
+import os 
+from tempfile import TemporaryDirectory
+import argparse
+
+def createfolders(args) -> None: 
+    '''
+    Creates folders for storing data, results, model weights. 
+    
+    Parameters
+    ----------
+    args
+        ArgParse object containing string paths to be created
+    
+    '''
+    
+    if not os.path.exists(args.datafolder):
+        os.makedirs(args.datafolder)
+        print(f'Created a folder at {args.datafolder}')
+        
+    if not os.path.exists(args.resultfolder):
+        os.makedirs(args.resultfolder)
+        print(f'Created a folder at {args.resultfolder}')
+        
+    if not os.path.exists(args.modelfolder):
+        os.makedirs(args.modelfolder)
+        print(f'Created a folder at {args.modelfolder}')
+
+
+
+def test_createfolders():
+    with TemporaryDirectory(dir = 'tmp/') as temp_dir:
+        parser = argparse.ArgumentParser()
+        #Structuture related values
+        parser.add_argument('--datafolder', type=str, default=os.path.join(temp_dir, 'Data/'), help='Path to where data will be saved during training.')
+        parser.add_argument('--resultfolder', type=str, default=os.path.join(temp_dir, 'Results/'), help='Path to where results will be saved during evaluation.')
+        parser.add_argument('--modelfolder', type=str, default=os.path.join(temp_dir, 'Experiments/'), help='Path to where model weights will be saved at the end of training.')
+        
+        args = parser.parse_args()
+        createfolders(args)
+        
+    return 

utils/dataloaders/svhn.py

@@ -0,0 +1,11 @@
+from torch.utils.data import Dataset
+
+class SVHN(Dataset):
+    def __init__(self):
+        super().__init__()
+    
+    def __len__(self):
+        return 
+    
+    def __getitem__(self, index):
+        return

utils/load_data.py

@@ -0,0 +1,5 @@
+from torch.utils.data import Dataset
+
+def load_data(dataset:str) -> Dataset:
+    
+    raise ValueError(f'Dataset: {dataset} not implemented. \nCheck the documentation for implemented metrics, or check your spelling')

utils/load_model.py

@@ -1,5 +1,9 @@
 import torch.nn as nn 
+from models import MagnusModel
 
 def load_model(modelname:str) -> nn.Module:
 
-    raise ValueError(f'Metric: {modelname} has not been implemented. \nCheck the documentation for implemented metrics, or check your spelling')
+    if modelname == 'MagnusModel':
+        return MagnusModel()
+    else:
+        raise ValueError(f'Model: {modelname} has not been implemented. \nCheck the documentation for implemented metrics, or check your spelling')

utils/models/__init__.py

@@ -0,0 +1 @@
+from .magnus_model import MagnusModel

utils/models/magnus_model.py

@@ -0,0 +1,8 @@
+import torch.nn as nn 
+
+class MagnusModel(nn.Module):
+    def __init__(self):
+        super().__init__()
+        
+    def forward(self, x):
+        return