AhamadShaik commited on
Commit
f4096b3
1 Parent(s): 717a6f0

Training in progress epoch 6

Browse files
Files changed (2) hide show
  1. README.md +8 -7
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,14 +14,14 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 0.0886
18
- - Train Dice Coef: 0.7856
19
- - Train Iou: 0.6535
20
- - Validation Loss: 0.0584
21
- - Validation Dice Coef: 0.8603
22
- - Validation Iou: 0.7566
23
  - Train Lr: 1e-04
24
- - Epoch: 5
25
 
26
  ## Model description
27
 
@@ -53,6 +53,7 @@ The following hyperparameters were used during training:
53
  | 0.0985 | 0.7610 | 0.6226 | 0.0632 | 0.8518 | 0.7440 | 1e-04 | 3 |
54
  | 0.0933 | 0.7745 | 0.6399 | 0.0627 | 0.8455 | 0.7351 | 1e-04 | 4 |
55
  | 0.0886 | 0.7856 | 0.6535 | 0.0584 | 0.8603 | 0.7566 | 1e-04 | 5 |
 
56
 
57
 
58
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 0.0831
18
+ - Train Dice Coef: 0.7971
19
+ - Train Iou: 0.6695
20
+ - Validation Loss: 0.0559
21
+ - Validation Dice Coef: 0.8621
22
+ - Validation Iou: 0.7596
23
  - Train Lr: 1e-04
24
+ - Epoch: 6
25
 
26
  ## Model description
27
 
 
53
  | 0.0985 | 0.7610 | 0.6226 | 0.0632 | 0.8518 | 0.7440 | 1e-04 | 3 |
54
  | 0.0933 | 0.7745 | 0.6399 | 0.0627 | 0.8455 | 0.7351 | 1e-04 | 4 |
55
  | 0.0886 | 0.7856 | 0.6535 | 0.0584 | 0.8603 | 0.7566 | 1e-04 | 5 |
56
+ | 0.0831 | 0.7971 | 0.6695 | 0.0559 | 0.8621 | 0.7596 | 1e-04 | 6 |
57
 
58
 
59
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:075a502ffb1230a869e651c81a65bc62ce2a77671a87977a1cabde87c8ccb9d4
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ddc2da57d9ebeed0be8b78a81bd6f7fe7534fb7c863bca4384bee2176d6466cd
3
  size 15133664