Spaces:
Sleeping
Sleeping
use steps and check
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ config = PPOConfig(
|
|
17 |
batch_size=1,
|
18 |
learning_rate=1e-5,
|
19 |
mini_batch_size=1,
|
20 |
-
|
21 |
)
|
22 |
|
23 |
# Create PPO Trainer
|
|
|
17 |
batch_size=1,
|
18 |
learning_rate=1e-5,
|
19 |
mini_batch_size=1,
|
20 |
+
steps=1 # Minimal epochs
|
21 |
)
|
22 |
|
23 |
# Create PPO Trainer
|