sebastiansarasti commited on
Commit
19d15fb
·
verified ·
1 Parent(s): bce9aa5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
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