Antigravity Agent commited on
Commit
f7cade8
·
1 Parent(s): 1ec970c

Deploy Fix: Disable interactive build

Browse files
Files changed (2) hide show
  1. models/liquid_ppo.py +2 -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 16 parallel environments.
73
- n_envs = 16
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(