Update README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,12 @@ tags:
|
|
6 |
|
7 |
## Model description
|
8 |
|
9 |
-
|
|
|
|
|
|
|
|
|
|
|
10 |
|
11 |
## Intended uses & limitations
|
12 |
|
@@ -14,14 +19,17 @@ More information needed
|
|
14 |
|
15 |
## Training and evaluation data
|
16 |
|
17 |
-
|
18 |
|
19 |
## Training procedure
|
20 |
|
21 |
### Training hyperparameters
|
22 |
|
23 |
The following hyperparameters were used during training:
|
24 |
-
|
|
|
|
|
|
|
25 |
| name | learning_rate | decay | beta_1 | beta_2 | epsilon | amsgrad | training_precision |
|
26 |
|----|-------------|-----|------|------|-------|-------|------------------|
|
27 |
|Adam|0.0010000000474974513|0.0|0.8999999761581421|0.9990000128746033|1e-07|False|float32|
|
|
|
6 |
|
7 |
## Model description
|
8 |
|
9 |
+
Metric learning aims to measure the similarity among data samples and to learn embedding models. In the context of metric learning to learn embedding models, the motivation is to embed inputs in an embedding space such that similar images
|
10 |
+
are close together in that space while dissimilar ones are far away.
|
11 |
+
|
12 |
+
These models once trained can produce embeddings for downstream systems where such similarity is useful.
|
13 |
+
|
14 |
+
Full credits to [Mat Kelcey](https://twitter.com/mat_kelcey) for this work.
|
15 |
|
16 |
## Intended uses & limitations
|
17 |
|
|
|
19 |
|
20 |
## Training and evaluation data
|
21 |
|
22 |
+
Trained and evaluated on [CIFAR-10](https://keras.io/api/datasets/cifar10/) dataset.
|
23 |
|
24 |
## Training procedure
|
25 |
|
26 |
### Training hyperparameters
|
27 |
|
28 |
The following hyperparameters were used during training:
|
29 |
+
- learning_rate: 0.001
|
30 |
+
- training_precision: float32
|
31 |
+
- num_epochs: 20
|
32 |
+
- Optimizer Info:
|
33 |
| name | learning_rate | decay | beta_1 | beta_2 | epsilon | amsgrad | training_precision |
|
34 |
|----|-------------|-----|------|------|-------|-------|------------------|
|
35 |
|Adam|0.0010000000474974513|0.0|0.8999999761581421|0.9990000128746033|1e-07|False|float32|
|