Spaces:
Runtime error
Runtime error
Antigravity Agent
commited on
Commit
·
3edc0c7
1
Parent(s):
dbef702
Deploy Fix: Disable interactive build
Browse files- Dockerfile +1 -1
- 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", "
|
|
|
|
| 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 =
|
| 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 |
-
|
| 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='
|
| 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
|