Antigravity Agent commited on
Commit
3edc0c7
·
1 Parent(s): dbef702

Deploy Fix: Disable interactive build

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. models/liquid_ppo.py +3 -3
Dockerfile CHANGED
@@ -24,4 +24,4 @@ COPY . .
24
 
25
  # Default command (can be overridden in Space settings)
26
  # Expects HF_TOKEN and REPO_ID env vars to be set in the Space
27
- CMD ["python", "train_hf.py", "--repo_id", "ylop/neuro-flyt-3d", "--steps", "100000"]
 
24
 
25
  # Default command (can be overridden in Space settings)
26
  # Expects HF_TOKEN and REPO_ID env vars to be set in the Space
27
+ CMD ["python", "train_hf.py", "--repo_id", "ylop/neuro-flyt-3d", "--steps", "500000"]
models/liquid_ppo.py CHANGED
@@ -70,11 +70,11 @@ def make_liquid_ppo(env, verbose=1):
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 1 environment to debug (DummyVecEnv)
73
- n_envs = 1
74
  env = make_vec_env(
75
  lambda: Drone3DEnv(render_mode=None, wind_scale=10.0, wind_speed=5.0),
76
  n_envs=n_envs,
77
- # vec_env_cls=SubprocVecEnv # Commented out to use DummyVecEnv for single process stability
78
  )
79
 
80
  # Create Model with optimized hyperparameters for A100
@@ -95,6 +95,6 @@ def make_liquid_ppo(env, verbose=1):
95
  gae_lambda=0.95,
96
  clip_range=0.2,
97
  policy_kwargs=policy_kwargs,
98
- device='cpu' # Force CPU to avoid CUDA/Multiprocessing issues and for better performance on small nets
99
  )
100
  return model
 
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 1 environment to debug (DummyVecEnv)
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,
77
+ vec_env_cls=SubprocVecEnv
78
  )
79
 
80
  # Create Model with optimized hyperparameters for A100
 
95
  gae_lambda=0.95,
96
  clip_range=0.2,
97
  policy_kwargs=policy_kwargs,
98
+ device='cuda' # Use GPU
99
  )
100
  return model