Finish run 2

Code/python/notebooks/hyperparameter_scan.py

@@ -3,8 +3,8 @@ import pathlib
 import pandas as pd
 from aiRNN import dataloader, models, losses
 
-MODE = 1
-DEVICE = "cuda"
+MODE = 2
+DEVICE = "cuda:2"
 
 
 # ------------------------------------------------------------
@@ -181,7 +181,7 @@ def run_mode_1():
 def run_mode_2():
     results = []
 
-    step = 1
+    step = 10
     base_ds = dataloader.SaveDataset(torch.load("dataset.pt"), step=step)
     train_ds, val_ds = split_dataset(base_ds)
 
@@ -228,7 +228,7 @@ def run_mode_2():
 def run_mode_3():
     results = []
 
-    step = 1
+    step = 10
     base_name = "LSTM"
     cls = models.ThreeInputLSTM
     hidden_size = 16