Bhavibond commited on
Commit
25f3d38
·
verified ·
1 Parent(s): 5c74aae

use steps and check

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ config = PPOConfig(
17
  batch_size=1,
18
  learning_rate=1e-5,
19
  mini_batch_size=1,
20
- total_ppo_epochs=1 # Minimal epochs
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