Update README.md
Browse files
README.md
CHANGED
|
@@ -66,3 +66,5 @@ $$
|
|
| 66 |
| Augmentations | Crop, flip, jitter, blur, rotation |
|
| 67 |
|
| 68 |
Trained using a cosine learning rate schedule with gradient clipping and automatic mixed precision (`torch.cuda.amp`)
|
|
|
|
|
|
|
|
|
| 66 |
| Augmentations | Crop, flip, jitter, blur, rotation |
|
| 67 |
|
| 68 |
Trained using a cosine learning rate schedule with gradient clipping and automatic mixed precision (`torch.cuda.amp`)
|
| 69 |
+
|
| 70 |
+
## Note: Model class will be available for usage once original repository becomes open-source (W.I.P)
|