fix: set log_dir to None for OnPolicyRunnerCTS init.
This commit is contained in:
@@ -198,7 +198,7 @@ def main(env_cfg: ManagerBasedRLEnvCfg | DirectRLEnvCfg | DirectMARLEnvCfg, agen
|
|||||||
elif agent_cfg.class_name == "DistillationRunner":
|
elif agent_cfg.class_name == "DistillationRunner":
|
||||||
runner = DistillationRunner(env, agent_cfg.to_dict(), log_dir=None, device=agent_cfg.device)
|
runner = DistillationRunner(env, agent_cfg.to_dict(), log_dir=None, device=agent_cfg.device)
|
||||||
elif agent_cfg.class_name == "OnPolicyRunnerCTS":
|
elif agent_cfg.class_name == "OnPolicyRunnerCTS":
|
||||||
runner = OnPolicyRunnerCTS(env, agent_cfg.to_dict(), log_dir=log_dir, device=agent_cfg.device)
|
runner = OnPolicyRunnerCTS(env, agent_cfg.to_dict(), log_dir=None, device=agent_cfg.device)
|
||||||
else:
|
else:
|
||||||
raise ValueError(f"Unsupported runner class: {agent_cfg.class_name}")
|
raise ValueError(f"Unsupported runner class: {agent_cfg.class_name}")
|
||||||
runner.load(resume_path)
|
runner.load(resume_path)
|
||||||
|
|||||||
Reference in New Issue
Block a user