Qingfeng Lan
Qingfeng Lan
Similar problem here. When I try to use WeightDropGRU and move it to GPU, I get an error: AttributeError: 'WeightDropGRU' object has no attribute '_flat_weights' Any idea how to solve...
Thank you for your detailed reply! I've emailed the author.
I encountered the same problem.
With Brax 0.9.3, PPO can only achieve 3000 in HalfCheetah with the implementation and hyper-parameters provided in the notebook, much lower than previous return (~8000).