To receive notifications about scheduled maintenance, please subscribe to the mailing-list gitlab-operations@sympa.ethz.ch. You can subscribe to the mailing-list at https://sympa.ethz.ch

Commit 4cd00f24 authored by Lukas Wolf's avatar Lukas Wolf
Browse files

add amplitude task

parent 0bb86b9a
......@@ -54,8 +54,8 @@ class Trainer:
elif config['task'] == 'gaze-reg':
logging.info("Loading coordinate task data")
data = np.load('./data/prepared/Position_task_with_dots_synchronised_' + config['preprocessing'] + '.npz')
elif config['task'] == 'angle-reg':
logging.info("Loading angle regression data")
elif config['task'] == 'angle-reg' or config['task'] == 'amplitude-reg':
logging.info(f"Loading {config['task']} regression data")
if config['dataset'] == 'calibration_task':
data = np.load('./data/prepared/Direction_task_with_dots_synchronised_' + config['preprocessing'] + '.npz')
else:
......
{
"cnn": {
"prosaccade-clf": {
"learning_rate": 1e-3,
"learning_rate": 1e-4,
"regularization": 0
},
"gaze-reg": {
......@@ -11,11 +11,15 @@
"angle-reg": {
"learning_rate": 1e-3,
"regularization": 0
},
"amplitude-reg": {
"learning_rate": 1e-3,
"regularization": 0
}
},
"xception": {
"prosaccade-clf": {
"learning_rate": 1e-3,
"learning_rate": 1e-4,
"regularization": 0
},
"gaze-reg": {
......@@ -25,11 +29,15 @@
"angle-reg": {
"learning_rate": 1e-3,
"regularization": 0
},
"amplitude-reg": {
"learning_rate": 1e-3,
"regularization": 0
}
},
},
"inception": {
"prosaccade-clf": {
"learning_rate": 1e-3,
"learning_rate": 1e-4,
"regularization": 0
},
"gaze-reg": {
......@@ -39,11 +47,15 @@
"angle-reg": {
"learning_rate": 1e-3,
"regularization": 0
},
"amplitude-reg": {
"learning_rate": 1e-3,
"regularization": 0
}
},
"pyramidal_cnn": {
"prosaccade-clf": {
"learning_rate": 1e-3,
"learning_rate": 1e-4,
"regularization": 0
},
"gaze-reg": {
......@@ -53,11 +65,15 @@
"angle-reg": {
"learning_rate": 1e-3,
"regularization": 0
},
"amplitude-reg": {
"learning_rate": 1e-3,
"regularization": 0
}
},
"eegnet": {
"prosaccade-clf": {
"learning_rate": 1e-3,
"learning_rate": 1e-4,
"regularization": 0
},
"gaze-reg": {
......@@ -67,6 +83,10 @@
"angle-reg": {
"learning_rate": 1e-3,
"regularization": 0
},
"amplitude-reg": {
"learning_rate": 1e-3,
"regularization": 0
}
}
}
\ No newline at end of file
......@@ -9,7 +9,9 @@ def log_config():
if config['task'] == 'gaze-reg':
logging.info("Running on the gaze regression task")
elif config['task'] == 'angle-reg':
logging.info("Running the angle regression task")
logging.info(f"Running the angle regression on {config['dataset']} task")
elif config['task'] == 'amplitude-reg':
logging.info(f"Running the amplitude regression on {config['dataset']} task")
else:
logging.info("Running the LR task")
logging.info("------------------------------------------------------------------------------------")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment