wufan commited on
Commit
906b977
1 Parent(s): 628ad0d

Update cfg_tiny.yaml

Browse files
Files changed (1) hide show
  1. cfg_tiny.yaml +3 -3
cfg_tiny.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_tiny.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_tiny
6
  max_seq_len: 1536
7
 
8
  load_pretrained: True
9
+ pretrained: './models/unimernet_tiny/unimernet_tiny.pth'
10
  tokenizer_config:
11
+ path: ./models/unimernet_tiny
12
 
13
  datasets:
14
  formula_rec_eval: