Preston Jiang

icon indicating a website link https://lpjiang97.github.io icon indicating an email [email protected]

icon company University of Washington icon location Seattle, WA icon location PhD student in Computer Science at the UW. Interested in theoretical neuroscience.

Results 4 issues of Preston Jiang

Slower performance on GPU than CPU with Zygote

15 comment

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...

GPU
gradient

[Bug Report] ViewSizeWrapper ignored with RGBImgPartialObsWrapper

1 comment

**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...

adapt moving mnist script to python3

Scripts running in Python3

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`)