pytorch - naming refactor - max_iters to n_steps

2026-02-26 22:31:32 +00:00 · 2023-08-04 13:45:21 +00:00
parent d17bf6350d
commit a3c6904fbc
5 changed files with 12 additions and 12 deletions
--- a/tests/freqai/conftest.py
+++ b/tests/freqai/conftest.py
@@ -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,
-                "n_epochs": None,
+                "n_epochs": 1,
            },
            "model_kwargs": {
                "hidden_dim": 32,