Thorben Frank issues

Repositories
Issues
Comments

Results 2 issues of


                                            Thorben Frank

Struggling to restore metadata on other device

Hello, I am trying to load metadata on a new device from a checkpoint via `CheckpointManager` API, but somehow struggle to find a solution. Below is a minimal example of...

`GraphsTuple` vs. batched `GraphsTuple` in `get_get_graph_padding_mask`

Hey, thanks for this great package. I realized that `jraph.get_graph_padding_mask` returns `jnp.array([False])` when applied to a non-batched `GraphsTuple`. I am wondering why this is? Would it be possible to check...