From cfc67cc68fc1a5a4dc568044a9d836c05153bb11 Mon Sep 17 00:00:00 2001
From: Joerg Martin <joerg.martin@ptb.de>
Date: Fri, 17 Dec 2021 12:55:17 +0100
Subject: [PATCH] Speeded up training on linear dataset

---
 EIVPackage/EIVData/linear.py                  | 4 ++--
 Experiments/configurations/eiv_linear.json    | 6 +++---
 Experiments/configurations/noneiv_linear.json | 4 ++--
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/EIVPackage/EIVData/linear.py b/EIVPackage/EIVData/linear.py
index 5dc363b..b182590 100644
--- a/EIVPackage/EIVData/linear.py
+++ b/EIVPackage/EIVData/linear.py
@@ -2,11 +2,11 @@ import torch
 import sys
 from torch.utils.data import TensorDataset, random_split
 
-total_number_of_datapoints = 10000
+total_number_of_datapoints = 2000
 input_range = [-1,1]
 slope = 1.0
 intercept = 0.0
-x_noise_strength = 0.1
+x_noise_strength = 0.05
 y_noise_strength = 0.1
 
 def get_normalization(*args):
diff --git a/Experiments/configurations/eiv_linear.json b/Experiments/configurations/eiv_linear.json
index 92c292e..a83fc15 100644
--- a/Experiments/configurations/eiv_linear.json
+++ b/Experiments/configurations/eiv_linear.json
@@ -2,7 +2,7 @@
 	"long_dataname": "linear",
 	"short_dataname": "linear",
 	"lr": 1e-3,
-	"batch_size": 100,
+	"batch_size": 64,
 	"test_batch_size": 800,
 	"number_of_epochs": 100,
 	"unscaled_reg": 10,
@@ -14,8 +14,8 @@
 	"eiv_prediction_number_of_batches": 10,
 	"init_std_y_list": [0.5],
 	"gamma": 0.5,
-	"hidden_layers": [1024, 1024, 1024, 1024],
-	"fixed_std_x": 0.1,
+	"hidden_layers": [128, 128, 128, 128],
+	"fixed_std_x": 0.05,
 	"seed_range": [0,10],
 	"gpu_number": 1
 }
diff --git a/Experiments/configurations/noneiv_linear.json b/Experiments/configurations/noneiv_linear.json
index 67c6ea4..1b2110a 100644
--- a/Experiments/configurations/noneiv_linear.json
+++ b/Experiments/configurations/noneiv_linear.json
@@ -2,7 +2,7 @@
 	"long_dataname": "linear",
 	"short_dataname": "linear",
 	"lr": 1e-3,
-	"batch_size": 100,
+	"batch_size": 64,
 	"test_batch_size": 800,
 	"number_of_epochs": 100,
 	"unscaled_reg": 10,
@@ -14,7 +14,7 @@
 	"noneiv_prediction_number_of_batches": 10,
 	"init_std_y_list": [0.5],
 	"gamma": 0.5,
-	"hidden_layers": [1024, 1024, 1024, 1024],
+	"hidden_layers": [128, 128, 128, 128],
 	"seed_range": [0,10],
 	"gpu_number": 1
 }
-- 
GitLab