torchrl.trainers.algorithms.configs.trainers.DQNTrainerConfig¶
- class torchrl.trainers.algorithms.configs.trainers.DQNTrainerConfig(collector: Any, total_frames: int, optim_steps_per_batch: int | None, loss_module: Any, optimizer: Any, logger: Any, save_trainer_file: Any, replay_buffer: Any, frame_skip: int = 1, clip_grad_norm: bool = True, clip_norm: float | None = None, progress_bar: bool = True, seed: int | None = None, save_trainer_interval: int = 10000, log_interval: int = 10000, create_env_fn: Any = None, value_network: Any = None, target_net_updater: Any = None, eps_init: float = 1.0, eps_end: float = 0.05, annealing_num_steps: int = 250000, async_collection: bool = False, log_timings: bool = False, _target_: str = 'torchrl.trainers.algorithms.configs.trainers._make_dqn_trainer')[source]¶
Configuration class for DQN (Deep Q-Network) trainer.