Merge pull request #8903 from Yinon-Polak/freqai-pytorch-bugfixes

Freqai pytorch bugfixes
This commit is contained in:
Robert Caulk
2023-08-15 16:48:44 +02:00
committed by GitHub
7 changed files with 62 additions and 79 deletions

View File

@@ -97,9 +97,9 @@ def mock_pytorch_mlp_model_training_parameters() -> Dict[str, Any]:
return {
"learning_rate": 3e-4,
"trainer_kwargs": {
"max_iters": 1,
"n_steps": None,
"batch_size": 64,
"max_n_eval_batches": 1,
"n_epochs": 1,
},
"model_kwargs": {
"hidden_dim": 32,