Skip to content

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

More bfloat16

parent a1329e4f
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 40 additions and 34 deletions
ygoai/rl/env.py
View file @ b0d45e40
...@@ -78,6 +78,7 @@ class EnvPreprocess(gym.Wrapper): ...@@ -78,6 +78,7 @@ class EnvPreprocess(gym.Wrapper):
def __init__(self, env, skip_mask): def __init__(self, env, skip_mask):
super().__init__(env) super().__init__(env)
self.num_envs = env.num_envs
self.skip_mask = skip_mask self.skip_mask = skip_mask
def reset(self, **kwargs): def reset(self, **kwargs):
......
ygoai/rl/jax/agent.py
View file @ b0d45e40
This diff is collapsed.
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