slm_lab/spec/benchmark/async_sac/async_sac_qbert.json
{
"async_sac_qbert": {
"agent": [{
"name": "SoftActorCritic",
"algorithm": {
"name": "SoftActorCritic",
"action_pdtype": "GumbelSoftmax",
"action_policy": "default",
"gamma": 0.99,
"training_start_step": 1000,
"training_frequency": 4
},
"memory": {
"name": "Replay",
"batch_size": 256,
"max_size": 200000,
"use_cer": false
},
"net": {
"type": "ConvNet",
"shared": false,
"conv_hid_layers": [
[32, 8, 4, 0, 1],
[64, 4, 2, 0, 1],
[32, 3, 1, 0, 1]
],
"fc_hid_layers": [256, 256],
"hid_layers_activation": "leakyrelu",
"init_fn": "orthogonal_",
"normalize": true,
"batch_norm": false,
"clip_grad_val": 0.5,
"use_same_optim": false,
"loss_spec": {
"name": "SmoothL1Loss"
},
"optim_spec": {
"name": "GlobalAdam",
"lr": 1e-4,
},
"lr_scheduler_spec": null,
"update_type": "polyak",
"update_frequency": 1,
"polyak_coef": 0.005,
"gpu": true
}
}],
"env": [{
"name": "QbertNoFrameskip-v4",
"frame_op": "concat",
"frame_op_len": 4,
"image_downsize": [64, 64],
"reward_scale": "sign",
"num_envs": 4,
"max_t": null,
"max_frame": 2e6
}],
"body": {
"product": "outer",
"num": 1
},
"meta": {
"distributed": "shared",
"log_frequency": 1000,
"eval_frequency": 1000,
"rigorous_eval": 0,
"max_session": 4,
"max_trial": 1,
}
},
"async_sac_qbert_small": {
"agent": [{
"name": "SoftActorCritic",
"algorithm": {
"name": "SoftActorCritic",
"action_pdtype": "GumbelSoftmax",
"action_policy": "default",
"gamma": 0.99,
"training_start_step": 10000,
"training_frequency": 1
},
"memory": {
"name": "Replay",
"batch_size": 256,
"max_size": 200000,
"use_cer": false
},
"net": {
"type": "ConvNet",
"shared": false,
"conv_hid_layers": [
[32, 8, 4, 0, 1],
[64, 4, 2, 0, 1],
[32, 3, 1, 0, 1]
],
"fc_hid_layers": [256, 256],
"hid_layers_activation": "leakyrelu",
"init_fn": "orthogonal_",
"normalize": true,
"batch_norm": false,
"clip_grad_val": 0.5,
"use_same_optim": false,
"loss_spec": {
"name": "SmoothL1Loss"
},
"optim_spec": {
"name": "GlobalAdam",
"lr": 1e-4,
},
"lr_scheduler_spec": null,
"update_type": "polyak",
"update_frequency": 1,
"polyak_coef": 0.005,
"gpu": true
}
}],
"env": [{
"name": "QbertNoFrameskip-v4",
"frame_op": "concat",
"frame_op_len": 4,
"image_downsize": [64, 64],
"reward_scale":null,
"num_envs": 4,
"max_t": null,
"max_frame": 1e7
}],
"body": {
"product": "outer",
"num": 1
},
"meta": {
"distributed": "shared",
"log_frequency": 1000,
"eval_frequency": 1000,
"rigorous_eval": 0,
"max_session": 4,
"max_trial": 1,
}
},
"async_sac_qbert_bignethighlr": {
"agent": [{
"name": "SoftActorCritic",
"algorithm": {
"name": "SoftActorCritic",
"action_pdtype": "GumbelSoftmax",
"action_policy": "default",
"gamma": 0.99,
"training_start_step": 1000,
"training_frequency": 4
},
"memory": {
"name": "Replay",
"batch_size": 512,
"max_size": 200000,
"use_cer": false
},
"net": {
"type": "ConvNet",
"shared": false,
"conv_hid_layers": [
[32, 8, 4, 0, 1],
[64, 4, 2, 0, 1],
[64, 3, 1, 0, 1]
],
"fc_hid_layers": [256, 256],
"hid_layers_activation": "relu",
"init_fn": "orthogonal_",
"normalize": true,
"batch_norm": false,
"clip_grad_val": 0.5,
"use_same_optim": false,
"loss_spec": {
"name": "SmoothL1Loss"
},
"optim_spec": {
"name": "GlobalAdam",
"lr": 2.5e-4,
},
"lr_scheduler_spec": null,
"update_type": "polyak",
"update_frequency": 1,
"polyak_coef": 0.005,
"gpu": true
}
}],
"env": [{
"name": "QbertNoFrameskip-v4",
"frame_op": "concat",
"frame_op_len": 4,
"reward_scale": "sign",
"num_envs": 4,
"max_t": null,
"max_frame": 2e6
}],
"body": {
"product": "outer",
"num": 1
},
"meta": {
"distributed": "shared",
"log_frequency": 1000,
"eval_frequency": 1000,
"rigorous_eval": 0,
"max_session": 6,
"max_trial": 1,
}
}
}