adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reduce mlp number of parameters for testing

Browse Source
This commit is contained in:
Yinon Polak 2023-03-13 20:18:26 +02:00
parent 918889a2bd
commit 366740885a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/freqai/test_freqai_interface.py
View File

@ -92,7 +92,7 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca,
"max_iters": 1, "max_iters": 1,
"batch_size": 64, "batch_size": 64,
"learning_rate": 3e-4, "learning_rate": 3e-4,
"max_n_eval_batches": None, "max_n_eval_batches": 1,
"model_kwargs": { "model_kwargs": {
"hidden_dim": 32, "hidden_dim": 32,
"dropout_percent": 0.2, "dropout_percent": 0.2,