| { | |
| "architectures": [ | |
| "Gr00tN1d7" | |
| ], | |
| "backbone_trainable_params_fp32": true, | |
| "color_jitter_params": { | |
| "brightness": 0.3, | |
| "contrast": 0.4, | |
| "hue": 0.08, | |
| "saturation": 0.5 | |
| }, | |
| "diffusion_model_cfg": { | |
| "attention_head_dim": 48, | |
| "dropout": 0.2, | |
| "final_dropout": true, | |
| "interleave_self_attention": true, | |
| "norm_type": "ada_norm", | |
| "num_attention_heads": 32, | |
| "num_layers": 16, | |
| "output_dim": 1024, | |
| "positional_embeddings": null | |
| }, | |
| "dtype": "float32", | |
| "extra_augmentation_config": null, | |
| "load_bf16": false, | |
| "model_name": "nvidia/Cosmos-Reason2-2B", | |
| "model_type": "Gr00tN1d7", | |
| "random_rotation_angle": null, | |
| "reproject_vision": false, | |
| "state_dropout_prob": 0.2, | |
| "transformers_version": "4.57.3", | |
| "tune_diffusion_model": true, | |
| "tune_llm": false, | |
| "tune_projector": true, | |
| "tune_visual": false, | |
| "use_relative_action": true | |
| } | |