Training in progress epoch 24

README.md

@@ -14,14 +14,14 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0521
-- Train Dice Coef: 0.8616
-- Train Iou: 0.7595
-- Validation Loss: 0.0483
-- Validation Dice Coef: 0.8822
-- Validation Iou: 0.7908
+- Train Loss: 0.0510
+- Train Dice Coef: 0.8634
+- Train Iou: 0.7624
+- Validation Loss: 0.0501
+- Validation Dice Coef: 0.8814
+- Validation Iou: 0.7899
 - Train Lr: 1e-04
-- Epoch: 23
+- Epoch: 24
 
 ## Model description
 
@@ -40,7 +40,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'Adam', 'learning_rate': 1e-04, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
+- optimizer: {'name': 'Adam', 'learning_rate': 5e-06, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
 - training_precision: float32
 
 ### Training results
@@ -71,6 +71,7 @@ The following hyperparameters were used during training:
 | 0.0515     | 0.8636          | 0.7625    | 0.0494          | 0.8816               | 0.7894         | 1e-04    | 21    |
 | 0.0518     | 0.8630          | 0.7615    | 0.0487          | 0.8836               | 0.7930         | 1e-04    | 22    |
 | 0.0521     | 0.8616          | 0.7595    | 0.0483          | 0.8822               | 0.7908         | 1e-04    | 23    |
+| 0.0510     | 0.8634          | 0.7624    | 0.0501          | 0.8814               | 0.7899         | 1e-04    | 24    |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:316be084fec0c2b2ab449e37ef7acd012bd79a98336d6e747832478fdea39cf9
+oid sha256:65eb3e10113f665420e6bea659be1ebf27ded61a603f9d97a3b86d0d1ba861f3
 size 15133664