Skip to content

Commit

Permalink
update qrdqn configs
Browse files Browse the repository at this point in the history
  • Loading branch information
wenzhangliu committed Jan 14, 2025
1 parent 9ff424f commit 670cb90
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions examples/qrdqn/qrdqn_configs/qrdqn_cartpole.yaml
Original file line number Diff line number Diff line change
@@ -1,3 +1,15 @@
dl_toolbox: "torch" # The deep learning toolbox. Choices: "torch", "mindspore", "tensorlayer"
project_name: "XuanCe_Benchmark"
logger: "tensorboard" # Choices: tensorboard, wandb.
wandb_user_name: "your_user_name"
render: True
render_mode: 'rgb_array' # Choices: 'human', 'rgb_array'.
fps: 50
test_mode: False
device: "cuda:0" # Choose an calculating device. PyTorch: "cpu", "cuda:0"; TensorFlow: "cpu"/"CPU", "gpu"/"GPU"; MindSpore: "CPU", "GPU", "Ascend", "Davinci".
distributed_training: False # Whether to use multi-GPU for distributed training.
master_port: '12355' # The master port for current experiment when use distributed training.

agent: "QRDQN"
env_name: "Classic Control"
env_id: "CartPole-v1"
Expand Down

0 comments on commit 670cb90

Please sign in to comment.