构建no-text数据集

This commit is contained in:
2026-02-20 14:13:25 +08:00
parent ee257bcb6c
commit d85cce8a52
5 changed files with 194 additions and 62 deletions

View File

@@ -67,6 +67,15 @@ ENDOSCOPE_TASK_CONFIGS = {
'text_max_length': 32,
'text_tokenizer_name': 'distilbert-base-uncased',
},
'endoscope_both_no_text': {
'dataset_dir': DATA_DIR + '/both-no-text',
'num_episodes': 3,
'episode_len': 400,
'camera_names': ['top'],
'state_dim': 2,
'action_dim': 2,
'use_text_instruction': False,
},
}
### Simulation envs fixed constants