reduce mlp number of parameters for testing

2023-03-13 20:18:26 +02:00
parent 918889a2bd
commit 366740885a
1 changed files with 1 additions and 1 deletions
--- a/tests/freqai/test_freqai_interface.py
+++ b/tests/freqai/test_freqai_interface.py
@@ -92,7 +92,7 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca,
            "max_iters": 1,
            "batch_size": 64,
            "learning_rate": 3e-4,
-            "max_n_eval_batches": None,
+            "max_n_eval_batches": 1,
            "model_kwargs": {
                "hidden_dim": 32,
                "dropout_percent": 0.2,