我正在尝试将一些cleanrl代码应用到机器人学习环境中。Cleanrl希望观察空间具有“shape”属性，但许多模拟机器人环境使用字典观察。有没有简单的方法告诉pytorch的SyncVectorEnv如何将字典观察展开为它可以使用的观察？
goqiplq21#
对于简单的情况，我想最好的方法是在SyncVectorEnv之前使用自定义 Package 器连接观察结果。
class DictConcatWrapper(gym.ObservationWrapper): def observation(self, obs): # (if dict, concatenate its elements here...) return obs
及以后版本：
env = ... env = DictConcatWrapper(env) env = SyncVectorEnv(env)
1条答案
按热度按时间goqiplq21#
对于简单的情况，我想最好的方法是在SyncVectorEnv之前使用自定义 Package 器连接观察结果。
及以后版本：