Wilka Carvalho
Wilka Carvalho
Following up on this. I'm trying to do this for an R2D2-like agent but it doesn't seem the work. Maybe it's because I have a custom learner? As a note,...
Thank you for the suggestion! I tried this and it didn't work :(
This is something I'd be very interested in! I ask because I generate hundreds of videos at a time for different runs and it would be nice to get nice,...
Also, as a note, I focus on object-interaction, so smoother frames when interacting with objects is what I'm looking for.
Here is a solution I found using `hk.lift`. I am very new to this whole vmap/haiku thing so I would really appreciate feedback on whether this is kosher or will...
Thank you for the prompt response! This solved my problem!
FYI to anyone that comes here. You need to add `split_rng=(not hk.running_init())` for `hk.vmap`
Question, the following for loop seems to be slowing down compilation signifcantly ``` functions = [hk.Linear(1) for i in range(8)] ``` Any thoughts on how to speed this up?
Hello Tom, my apologies for the late response. I'm finding that this has a long compile time when used to create parameters for the DQN loss objective in the ACME...
I think I figured out the problem. Inside an RNN, this is **SLOWER** than just using a for loop @tomhennigan