Preston Jiang

Slower performance on GPU than CPU with Zygote

15

I'm experiencing some slower performances when taking gradients on `CuArray`s, than normal CPU `Array`s, using Zygote. Here's an MWE (the Tullio operations implement a mixture linear regression): ```julia using Tullio...

[Bug Report] ViewSizeWrapper ignored with RGBImgPartialObsWrapper

1

**Describe the bug** When wrapped after the `RGBImgPartialObsWrapper`, the `agent_view_size` set in `ViewSizeWrapper` is ignored. **Code example** ``` env = gym.make("MiniGrid-Empty-8x8-v0") env = RGBImgPartialObsWrapper(ViewSizeWrapper(env, agent_view_size=5)) env = ImgObsWrapper(env) obs, info...

This modifies the scripts to run in Python 3.8. Requires a beta version of `pygame` to run (`pip install pygame==2.0.0.dev6`)

Duplicates in `num_neurons` in 34 session setting causing stitcher reuse

1

Hi authors, I noticed that in the 34 session setting, there are sessions that have the same number of neurons. Specifically, ``` Duplicate number of neurons found: ibl-foundation-model/f140a2ec-fd49-4814-994a-fe3476f14e66_aligned ibl-foundation-model/41872d7f-75cb-4445-bb1a-132b354c44f0_aligned Duplicate...

Hi all, I'm very interested in this work and would like to try the model on other sessions from the IBL dataset. It seems like the `src/data_loading_examples.py` and `notebooks/data_loading_examples.ipynb` files...

Typo on the number of time bins in the paper?

1

Hi all, In Section 4.1 of the paper, it is noted > We bin the neural activity using 20ms windows and we fix the trial-length to 2 seconds (200 time...