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
|