mirror of
https://github.com/wassname/curl.git
synced 2026-08-21 11:14:16 +08:00
9 lines
389 B
Bash
Executable File
9 lines
389 B
Bash
Executable File
CUDA_VISIBLE_DEVICES=5 python train.py \
|
|
--domain_name cartpole \
|
|
--task_name swingup \
|
|
--encoder_type pixel \
|
|
--action_repeat 8 \
|
|
--save_tb --pre_transform_image_size 100 --image_size 84 \
|
|
--work_dir ./tmp/cartpole \
|
|
--agent curl_sac --frame_stack 3 \
|
|
--seed -1 --critic_lr 1e-3 --actor_lr 1e-3 --eval_freq 10000 --batch_size 128 --num_train_steps 1000000 |