DQN agent for an environment with continuous actions
MishraIN opened this issue · 1 comments
MishraIN commented
I am trying to implement DQN using the continuous cartpole which came packaged with MBRL pets. When I am trying to discretized and flattened the action space I am getting error "AttributeError: 'float' object has no attribute 'squeeze'" in the last line of the code,
env= cartpole_env.CartPoleEnv()
import space_wrappers
wrapped = space_wrappers.DiscretizedActionWrapper(env, 2)
env = space_wrappers.FlattenedActionWrapper(wrapped)
env.reset()
env.step(1)