You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Some attrs like cumsum_nbranches have different types in different Modules.
Also we should make sure that the attributes are ideally numpy arrays not jax arrays.
Some attrs like
cumsum_nbranches
have different types in different Modules.Also we should make sure that the attributes are ideally numpy arrays not jax arrays.
fyi @michaeldeistler
The text was updated successfully, but these errors were encountered: