构建no-text数据集
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user