File size: 848 Bytes
fe5c524 68d15c4 fe5c524 5c583ae fe5c524 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
{
"activation_function": "gelu",
"architectures": [
"TrajectoryTransformerModel"
],
"N": 100,
"action_weight": 5,
"attn_pdrop": 0.1,
"batch_size": 256,
"discount": 0.99,
"bos_token_id": 50256,
"embd_pdrop": 0.1,
"learning_rate": 0.0006,
"lr_decay": true,
"eos_token_id": 50256,
"initializer_range": 0.02,
"layer_norm_epsilon": 1e-05,
"model_type": "trajectory_transformer",
"n_ctx": 1024,
"n_embd": 128,
"n_epochs_ref": 50,
"n_head": 4,
"n_layer": 4,
"n_saves": 3,
"resid_pdrop": 0.1,
"reward_weight": 1,
"seed": 42,
"summary_activation": null,
"summary_first_dropout": 0.1,
"summary_proj_to_labels": true,
"summary_type": "cls_index",
"summary_use_proj": true,
"vocab_size": 100,
"step": 1,
"subsampled_sequence_length": 10,
"termination_penalty": -100,
"value_weight": 1
} |