Update train_model.py

scripts/train_model.py

@@ -170,12 +170,12 @@ if __name__ == '__main__':
                 signals=np.array(training_df['signal']),
                 labels=np.array(training_df['azimuth']),
                 sounds=training_sounds,
-                n_samples=n_samples,
+                n_samples=None,
                 n_augment=1)
             test_signals, test_labels = generate_augmented_labelled_dataset(signals=np.array(test_df['signal']),
                                                                             labels=np.array(test_df['azimuth']),
                                                                             sounds=test_sounds,
-                                                                            n_samples=n_samples,
+                                                                            n_samples=None,
                                                                             n_augment=1)
 
             if exp_config['model_name'] == 'left_center_right':