reduce mlp number of parameters for testing

This commit is contained in:
Yinon Polak 2023-03-13 20:18:26 +02:00
parent 918889a2bd
commit 366740885a

View File

@ -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,