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
Hide 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__": ...@@ -585,7 +585,7 @@ if __name__ == "__main__":
), ),
every_k_schedule=1, 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( agent_state = TrainState.create(
apply_fn=None, apply_fn=None,
params=params, params=params,
......
scripts/ppo.py
View file @ 63aac472
...@@ -595,7 +595,7 @@ if __name__ == "__main__": ...@@ -595,7 +595,7 @@ if __name__ == "__main__":
), ),
every_k_schedule=1, 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( agent_state = TrainState.create(
apply_fn=None, apply_fn=None,
params=params, 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