vdprabhu commited on
Commit
b0ba0b9
·
1 Parent(s): 7aae728

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -3
README.md CHANGED
@@ -6,7 +6,12 @@ tags:
6
 
7
  ## Model description
8
 
9
- More information needed
 
 
 
 
 
10
 
11
  ## Intended uses & limitations
12
 
@@ -14,14 +19,17 @@ More information needed
14
 
15
  ## Training and evaluation data
16
 
17
- More information needed
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|