diff --git a/TrainerClass.py b/TrainerClass.py index 7f349f1..de8a750 100644 --- a/TrainerClass.py +++ b/TrainerClass.py @@ -191,9 +191,9 @@ class eNoseTrainer: # maxpool 2 o 3 => 2 | 2/15 def build_model_conv1D(config, input_shape, output_dim): model = keras.Sequential([ - layers.Conv1D(filters=config['filters'], kernel_size=config['kernel_l1'], stride=config['kernel_l1']//2, activation='relu', padding='causal', input_shape=input_shape), + layers.Conv1D(filters=config['filters'], kernel_size=config['kernel_l1'], strides=config['kernel_l1']//2, activation='relu', padding='causal', input_shape=input_shape), layers.MaxPooling1D(pool_size=config['pool_size']), - layers.Conv1D(filters=config['filters'] * 2, kernel_size=config['kernel_l2'], stride=config['kernel_l2']//2, activation='relu', padding='causal'), + layers.Conv1D(filters=config['filters'] * 2, kernel_size=config['kernel_l2'], strides=config['kernel_l2']//2, activation='relu', padding='causal'), layers.MaxPooling1D(pool_size=config['pool_size']), layers.Flatten(), layers.Dense(config['dense_units'], activation='relu'),