Spaces:
Runtime error
Runtime error
Antigravity Agent
commited on
Commit
·
1ec970c
1
Parent(s):
f7709b7
Deploy Fix: Disable interactive build
Browse files- train_hf.py +2 -2
train_hf.py
CHANGED
|
@@ -22,7 +22,7 @@ def train_hf(repo_id, token, total_timesteps=500000):
|
|
| 22 |
# Checkpoint Callback
|
| 23 |
checkpoint_callback = CheckpointCallback(
|
| 24 |
save_freq=50000,
|
| 25 |
-
save_path='
|
| 26 |
name_prefix='liquid_ppo_drone'
|
| 27 |
)
|
| 28 |
|
|
@@ -30,7 +30,7 @@ def train_hf(repo_id, token, total_timesteps=500000):
|
|
| 30 |
model.learn(total_timesteps=total_timesteps, callback=checkpoint_callback)
|
| 31 |
|
| 32 |
# Save Final Model
|
| 33 |
-
model_path = "liquid_ppo_drone_final.zip"
|
| 34 |
model.save(model_path)
|
| 35 |
print(f"Model saved to {model_path}")
|
| 36 |
|
|
|
|
| 22 |
# Checkpoint Callback
|
| 23 |
checkpoint_callback = CheckpointCallback(
|
| 24 |
save_freq=50000,
|
| 25 |
+
save_path='/tmp/checkpoints/',
|
| 26 |
name_prefix='liquid_ppo_drone'
|
| 27 |
)
|
| 28 |
|
|
|
|
| 30 |
model.learn(total_timesteps=total_timesteps, callback=checkpoint_callback)
|
| 31 |
|
| 32 |
# Save Final Model
|
| 33 |
+
model_path = "/tmp/liquid_ppo_drone_final.zip"
|
| 34 |
model.save(model_path)
|
| 35 |
print(f"Model saved to {model_path}")
|
| 36 |
|