Skip to content

Y
ygo-agent
Commit 63aac472 authored by sbl1996@126.com's avatar sbl1996@126.com
Browse files
Options

Allow max 10 nans

parent 4dd1538c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
scripts/impala.py
View file @ 63aac472
......@@ -585,7 +585,7 @@ if __name__ == "__main__":
),
every_k_schedule=1,
)
tx = optax.apply_if_finite(tx, max_consecutive_errors=3)
tx = optax.apply_if_finite(tx, max_consecutive_errors=10)
agent_state = TrainState.create(
apply_fn=None,
params=params,
......
scripts/ppo.py
View file @ 63aac472
......@@ -595,7 +595,7 @@ if __name__ == "__main__":
),
every_k_schedule=1,
)
tx = optax.apply_if_finite(tx, max_consecutive_errors=3)
tx = optax.apply_if_finite(tx, max_consecutive_errors=10)
agent_state = TrainState.create(
apply_fn=None,
params=params,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment