Configuration: config.py

Arguments

The Arguments class contains all parameters of the training process, including environment setup, model training, model evaluation, and resource allocation. It provides users an unified interface to customize the training process.

The class should be initialized at the start of the training process. For example,

from elegantrl.train.config import Arguments
from elegantrl.agents.AgentPPO import AgentPPO
from elegantrl.train.config import build_env
import gym

args = Arguments(build_env('Pendulum-v1'), AgentPPO())

The full list of parameters in Arguments:

class elegantrl.train.config.Arguments(agent_class=None, env=None, env_func=None, env_args: Optional[dict] = None)[source]
if_allow_break

Arguments for evaluate

if_use_old_traj

Arguments for device

learner_gpus

Arguments for evaluate

Environment registration

elegantrl.train.config.build_env(env=None, env_func=None, env_args=None)[source]

Utils

elegantrl.train.config.kwargs_filter(func, kwargs: dict)[source]