Spaces:
Runtime error
Runtime error
Antigravity Agent
commited on
Commit
·
f7cade8
1
Parent(s):
1ec970c
Deploy Fix: Disable interactive build
Browse files- models/liquid_ppo.py +2 -2
- train_hf.py +2 -0
models/liquid_ppo.py
CHANGED
|
@@ -69,8 +69,8 @@ def make_liquid_ppo(env, verbose=1):
|
|
| 69 |
"""
|
| 70 |
# Parallel Environments for High-Performance Training
|
| 71 |
# A100/A10G are data hungry. We need to run physics on many CPU cores to feed them.
|
| 72 |
-
# We will use
|
| 73 |
-
n_envs =
|
| 74 |
env = make_vec_env(
|
| 75 |
lambda: Drone3DEnv(render_mode=None, wind_scale=10.0, wind_speed=5.0),
|
| 76 |
n_envs=n_envs,
|
|
|
|
| 69 |
"""
|
| 70 |
# Parallel Environments for High-Performance Training
|
| 71 |
# A100/A10G are data hungry. We need to run physics on many CPU cores to feed them.
|
| 72 |
+
# We will use 4 parallel environments to prevent OOM/Shm issues on Space
|
| 73 |
+
n_envs = 4
|
| 74 |
env = make_vec_env(
|
| 75 |
lambda: Drone3DEnv(render_mode=None, wind_scale=10.0, wind_speed=5.0),
|
| 76 |
n_envs=n_envs,
|
train_hf.py
CHANGED
|
@@ -17,7 +17,9 @@ def train_hf(repo_id, token, total_timesteps=500000):
|
|
| 17 |
|
| 18 |
# Create Optimized Model (Parallel Envs + A100 Tuning)
|
| 19 |
# Note: make_liquid_ppo now handles env creation internally for parallelism
|
|
|
|
| 20 |
model = make_liquid_ppo(None, verbose=1)
|
|
|
|
| 21 |
|
| 22 |
# Checkpoint Callback
|
| 23 |
checkpoint_callback = CheckpointCallback(
|
|
|
|
| 17 |
|
| 18 |
# Create Optimized Model (Parallel Envs + A100 Tuning)
|
| 19 |
# Note: make_liquid_ppo now handles env creation internally for parallelism
|
| 20 |
+
print("Creating Liquid PPO Model...")
|
| 21 |
model = make_liquid_ppo(None, verbose=1)
|
| 22 |
+
print("Model created successfully.")
|
| 23 |
|
| 24 |
# Checkpoint Callback
|
| 25 |
checkpoint_callback = CheckpointCallback(
|