Commit a1329e4f authored by sbl1996@126.com's avatar sbl1996@126.com

Fix

parent 72a1fd28
......@@ -82,6 +82,7 @@ class CardEncoder(nn.Module):
channels: int = 128
dtype: Optional[jnp.dtype] = None
param_dtype: jnp.dtype = jnp.float32
oppo_info: bool = False
version: int = 0
@nn.compact
......
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