Update README.md
Browse files
README.md
CHANGED
|
@@ -63,10 +63,15 @@ class ModelColorization(nn.Module, PyTorchModelHubMixin):
|
|
| 63 |
The model was trained using PyTorch for 5 epochs. Here are the training and validation losses observed during the training:
|
| 64 |
|
| 65 |
Epoch 1: Training Loss: 0.0063, Validation Loss: 0.0042
|
|
|
|
| 66 |
Epoch 2: Training Loss: 0.0036, Validation Loss: 0.0035
|
|
|
|
| 67 |
Epoch 3: Training Loss: 0.0032, Validation Loss: 0.0032
|
|
|
|
| 68 |
Epoch 4: Training Loss: 0.0030, Validation Loss: 0.0030
|
|
|
|
| 69 |
Epoch 5: Training Loss: 0.0029, Validation Loss: 0.0030
|
|
|
|
| 70 |
The model demonstrated continuous improvement in reducing both training and validation loss over the epochs.
|
| 71 |
|
| 72 |
### Usage
|
|
|
|
| 63 |
The model was trained using PyTorch for 5 epochs. Here are the training and validation losses observed during the training:
|
| 64 |
|
| 65 |
Epoch 1: Training Loss: 0.0063, Validation Loss: 0.0042
|
| 66 |
+
|
| 67 |
Epoch 2: Training Loss: 0.0036, Validation Loss: 0.0035
|
| 68 |
+
|
| 69 |
Epoch 3: Training Loss: 0.0032, Validation Loss: 0.0032
|
| 70 |
+
|
| 71 |
Epoch 4: Training Loss: 0.0030, Validation Loss: 0.0030
|
| 72 |
+
|
| 73 |
Epoch 5: Training Loss: 0.0029, Validation Loss: 0.0030
|
| 74 |
+
|
| 75 |
The model demonstrated continuous improvement in reducing both training and validation loss over the epochs.
|
| 76 |
|
| 77 |
### Usage
|