| { | |
| "dit_core_only_unexpected_trainable_params": 0, | |
| "groups": { | |
| "_groot_model.action_head.model": { | |
| "total": 550386688, | |
| "trainable": 550386688 | |
| }, | |
| "_groot_model.action_head.vl_self_attention": { | |
| "total": 201433088, | |
| "trainable": 201433088 | |
| }, | |
| "_groot_model.action_head.vlln": { | |
| "total": 4096, | |
| "trainable": 4096 | |
| }, | |
| "sequence_mgd_head": { | |
| "total": 0, | |
| "trainable": 0 | |
| } | |
| }, | |
| "num_learnable_params": 1068806144, | |
| "num_total_params": 2724163520, | |
| "other_trainable_params": 1068806144, | |
| "policy_type": "groot", | |
| "selected_prefixes": [], | |
| "trainable_mode": null, | |
| "trainable_ratio": 0.3923428737493702 | |
| } |