YufeiWeng commited on
Commit
cec08a3
·
verified ·
1 Parent(s): e4ef7e3

Model save

Browse files
Files changed (3) hide show
  1. README.md +2 -4
  2. config.json +0 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -1,8 +1,6 @@
1
  ---
2
  base_model: microsoft/dit-base-finetuned-rvlcdip
3
  tags:
4
- - image-classification
5
- - vision
6
  - generated_from_trainer
7
  metrics:
8
  - f1
@@ -18,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [microsoft/dit-base-finetuned-rvlcdip](https://huggingface.co/microsoft/dit-base-finetuned-rvlcdip) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0385
22
- - F1: 0.6332
23
 
24
  ## Model description
25
 
 
1
  ---
2
  base_model: microsoft/dit-base-finetuned-rvlcdip
3
  tags:
 
 
4
  - generated_from_trainer
5
  metrics:
6
  - f1
 
16
 
17
  This model is a fine-tuned version of [microsoft/dit-base-finetuned-rvlcdip](https://huggingface.co/microsoft/dit-base-finetuned-rvlcdip) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - F1: 0.9705
20
+ - Loss: 0.0391
21
 
22
  ## Model description
23
 
config.json CHANGED
@@ -48,7 +48,6 @@
48
  3,
49
  6
50
  ],
51
- "problem_type": "regression",
52
  "reshape_hidden_states": true,
53
  "semantic_loss_ignore_index": 255,
54
  "stage_names": [
 
48
  3,
49
  6
50
  ],
 
51
  "reshape_hidden_states": true,
52
  "semantic_loss_ignore_index": 255,
53
  "stage_names": [
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:44b81663d616a9773718629abc17b9386affbd294c4a75a1e834cc952ba42ad3
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af3caba0c294e48c3fea421d968ca9847e83b8657b6f17547a3e4e9561bf509f
3
  size 5368