rail-berkeley/rlkit

Question about VAEPolicy in rlkit.torch.sac.policies

Closed this issue · 2 comments

HYDesmondLiu commented

In UWAC paper, they did from rlkit.torch.sac.policies import TanhGaussianPolicy, MakeDeterministic, VAEPolicy in examples/UWAC_hdf5_d4rl.py.

However, even in rlkit repository there is no such class. May I know where can I get the latest version of the repository?

anair13 commented

Yes, it looks like it is probably based off of this repo (which extends rlkit): https://github.com/rail-berkeley/d4rl_evaluations/blob/master/bear/rlkit/torch/sac/policies.py

HYDesmondLiu commented

Thank you @anair13 for the pointer, now it makes sense.

  • 👍1