具有具有輸入參數的環境的 Tf 代理並行 Py 環境

Question

假設您有一個具有輸入參數的環境：例如，要創建一個您將使用的實例

env_instance = MyEnv(var_1=3, var_2=5, ...)

現在假設您想使用環境“MyEnv”創建一個 parallel_py_environment ？ 由於您需要輸入參數，因此不能使用

tf_py_environment.TFPyEnvironment(parallel_py_environment.ParallelPyEnvironment([MyEnv]*int(n_envs)))

Answer 1

解決方法是創建一個超類：

class MyEnvPar(MyEnv):
    def __init__(self):
        super().__init__(var_1=3, var_2=5)

然后你可以使用

tf_py_environment.TFPyEnvironment(parallel_py_environment.ParallelPyEnvironment([MyEnvPar]*int(n_envs)))