diff --git a/code/train.py b/code/train.py index 7410741..6108a20 100644 --- a/code/train.py +++ b/code/train.py @@ -31,10 +31,10 @@ def main(argv): index = int(arg) baseline_configurations = Utility.create_configuration([Conf.baseline_var_names], [Conf.conf_baseline]) - experiment_configurations = Utility.create_configuration([Conf.hyperparameters]*5, + experiment_configurations = Utility.create_configuration([Conf.hyperparameters]*4, [Conf.conf_vector, Conf.conf_tensor_standard, - Conf.conf_tensor_linearity, + # Conf.conf_tensor_linearity, Conf.conf_tensor_resolution, Conf.conf_tensor_initialization])