wufan commited on
Commit
628ad0d
·
verified ·
1 Parent(s): ec0123f

Update cfg_base.yaml

Browse files
Files changed (1) hide show
  1. cfg_base.yaml +3 -3
cfg_base.yaml CHANGED
@@ -2,13 +2,13 @@ model:
2
  arch: unimernet
3
  model_type: unimernet
4
  model_config:
5
- model_name: MODEL_DIR
6
  max_seq_len: 1536
7
 
8
  load_pretrained: True
9
- pretrained: 'MODEL_DIR/unimernet_base.pth'
10
  tokenizer_config:
11
- path: MODEL_DIR
12
 
13
  datasets:
14
  formula_rec_eval:
 
2
  arch: unimernet
3
  model_type: unimernet
4
  model_config:
5
+ model_name: ./models/unimernet_base
6
  max_seq_len: 1536
7
 
8
  load_pretrained: True
9
+ pretrained: './models/unimernet_base/unimernet_base.pth'
10
  tokenizer_config:
11
+ path: ./models/unimernet_base
12
 
13
  datasets:
14
  formula_rec_eval: