train
parameter to false
in agent_cfg.yaml
.
This would run the agent in the training environment by utilizing the trained policy.
task
as GRID-CustomEnv-v0
and specifying the environment in the scene_cfg.yaml
enables users to use the trained policy in diverse environments.
A sample agent_cfg.yaml
file for inference is shown below: