jalf/TIR_PROJ
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Sheep training flock _ improver

Browse Source
This commit is contained in:
Johnny Fernandes
2026-04-24 23:27:05 +01:00
parent b3251fcca3
commit 0716c6c3c8
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/smoke_test.py
+1 -1
View File
@@ -110,7 +110,7 @@ def train_stage(n_sheep, steps, n_envs=4, prev_model=None, prev_vecnorm=None):
model = PPO( model = PPO(
"MlpPolicy", vn, "MlpPolicy", vn,
learning_rate=3e-4, n_steps=2048, batch_size=256, n_epochs=10, learning_rate=3e-4, n_steps=2048, batch_size=256, n_epochs=10,
gamma=0.995, gae_lambda=0.95, clip_range=0.2, ent_coef=0.02, gamma=0.995, gae_lambda=0.95, clip_range=0.2, ent_coef=0.005,
vf_coef=0.5, max_grad_norm=0.5, vf_coef=0.5, max_grad_norm=0.5,
policy_kwargs=dict(net_arch=[256, 256]), policy_kwargs=dict(net_arch=[256, 256]),
verbose=1, verbose=1,
training/train.py
+1 -1
View File
@@ -332,7 +332,7 @@ def main():
gamma = 0.995, gamma = 0.995,
gae_lambda = 0.95, gae_lambda = 0.95,
clip_range = 0.2, clip_range = 0.2,
ent_coef = 0.02, ent_coef = 0.005,
vf_coef = 0.5, vf_coef = 0.5,
max_grad_norm = 0.5, max_grad_norm = 0.5,
policy_kwargs = dict(net_arch=[256, 256]), policy_kwargs = dict(net_arch=[256, 256]),