Skip to content

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

Fix

parent 77492ca0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ygoai/rl/jax/agent.py
View file @ ef6e50ed
...@@ -265,7 +265,7 @@ class Encoder(nn.Module): ...@@ -265,7 +265,7 @@ class Encoder(nn.Module):
if f_cards is None: if f_cards is None:
fs_g_card.append(None) fs_g_card.append(None)
continue continue
name = 'g_card_embed' if i == 0 else 'g_g_card_embed' name = 'g_g_card_embed' if i == 0 else 'g_card_embed'
g_card_embed = self.param( g_card_embed = self.param(
name, name,
lambda key, shape, dtype: jax.random.normal(key, shape, dtype) * 0.02, lambda key, shape, dtype: jax.random.normal(key, shape, dtype) * 0.02,
......
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