| { | |
| "action_dim": 7, | |
| "action_head": "continuousactionhead", | |
| "batch_size": 256, | |
| "dataset_path": "/network/scratch/o/ozgur.aslan/libero_td/libero_10_dinov3_101469.pt", | |
| "dataset_repo_id": "lerobot/libero_10_image", | |
| "depth": 6, | |
| "device": "cuda", | |
| "dim_head": 64, | |
| "dino_config": { | |
| "img_size": 224, | |
| "model_name": "facebook/dinov3-vits16-pretrain-lvd1689m", | |
| "processor_name": "facebook/dinov3-vits16-pretrain-lvd1689m" | |
| }, | |
| "dtype_str": "bfloat16", | |
| "frozen_modules": [ | |
| "image_encoder", | |
| "text_encoder", | |
| "text_tokenizer" | |
| ], | |
| "hf_user_id": "ozgraslan", | |
| "hidden_dim": 512, | |
| "learning_rate": 0.0003, | |
| "log_freq": 100, | |
| "mlp_dim": 2048, | |
| "num_action_chunks": 32, | |
| "num_heads": 8, | |
| "num_input_tokens": 233, | |
| "num_iters": 30000, | |
| "num_state_tokens": 1, | |
| "patch_dim": 384, | |
| "policy_name": "d3sr_30kit_hid512_depth6_bs256_bf16_grp", | |
| "projection_layer": "dinoreglinearprojectionlayer", | |
| "push_to_hf": true, | |
| "reg_dim": 384, | |
| "save_path": "/network/scratch/o/ozgur.aslan/simple_grp", | |
| "seed": 0, | |
| "state_dim": 8, | |
| "text_dim": 512, | |
| "text_max_length": 32, | |
| "text_model_name": "google-t5/t5-small", | |
| "text_padding": "max_length", | |
| "torch_compile": false, | |
| "use_lr_scheduler": true, | |
| "use_wandb": true, | |
| "wproject": "sgrp-libero10" | |
| } |