| { | |
| "policy_config": { | |
| "auxiliary_head_kwargs": [ | |
| { | |
| "enable": true | |
| } | |
| ], | |
| "from": { | |
| "model": null, | |
| "weights": null | |
| }, | |
| "name": "rocket_minecraft", | |
| "pi_head_kwargs": { | |
| "temperature": 1.0 | |
| }, | |
| "policy_kwargs": { | |
| "attention_heads": 8, | |
| "attention_mask_style": "clipped_causal", | |
| "attention_memory_size": 256, | |
| "backbone_kwargs": { | |
| "accept_segment": true, | |
| "img_shape": [ | |
| 224, | |
| 224, | |
| 3 | |
| ], | |
| "name": "EFFICIENTNET", | |
| "pooling": false, | |
| "version": "efficientnet-b0" | |
| }, | |
| "hidsize": 1024, | |
| "init_norm_kwargs": { | |
| "batch_norm": false, | |
| "group_norm_groups": 1 | |
| }, | |
| "n_recurrence_layers": 4, | |
| "only_img_input": true, | |
| "pointwise_ratio": 4, | |
| "pointwise_use_activation": false, | |
| "recurrence_is_residual": true, | |
| "recurrence_type": "transformer", | |
| "timesteps": 128, | |
| "use_pointwise_layer": true, | |
| "use_pre_lstm_ln": false, | |
| "word_dropout": 0.0 | |
| }, | |
| "policy_name": "ROCKET" | |
| }, | |
| "weights_dict": null | |
| } |