Open agrawalraj opened 4 months ago
Thank you for creating this cool dataset! I would like to use the RL algorithms in TorchRL but that requires that Webarena be written in terms of TensorDict. TorchRL supports many simulation backends such as gym based on this note https://pytorch.org/rl/reference/generated/torchrl.envs.GymEnv.html#torchrl.envs.GymEnv.
TensorDict
gym
It seems like Webarena is based on gym too so it seems plausible to use TorchRL with Webarena. Any thoughts / suggestions here?
I would be happy to submit a PR to add TorchRL support if it's not currently possible.
Thank you for creating this cool dataset! I would like to use the RL algorithms in TorchRL but that requires that Webarena be written in terms of
TensorDict
. TorchRL supports many simulation backends such asgym
based on this note https://pytorch.org/rl/reference/generated/torchrl.envs.GymEnv.html#torchrl.envs.GymEnv.It seems like Webarena is based on
gym
too so it seems plausible to use TorchRL with Webarena. Any thoughts / suggestions here?I would be happy to submit a PR to add TorchRL support if it's not currently possible.