Added example params for batch_all and batch_hard.

omoindrot · Oct 22, 2018 · e2d2299 · e2d2299
1 parent ff1b415
commit e2d2299
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/experiments/.gitkeep b/experiments/.gitkeep
diff --git a/experiments/base_model/params.json → experiments/batch_all/params.json b/experiments/base_model/params.json → experiments/batch_all/params.json
diff --git a/experiments/batch_hard/params.json b/experiments/batch_hard/params.json
@@ -0,0 +1,21 @@
+{
+    "learning_rate": 1e-4,
+    "batch_size": 64,
+    "num_epochs": 20,
+
+    "num_channels": 32,
+    "use_batch_norm": false,
+    "bn_momentum": 0.9,
+    "margin": 0.5,
+    "embedding_size": 64,
+    "triplet_strategy": "batch_hard",
+    "squared": false,
+
+    "image_size": 28,
+    "num_labels": 10,
+    "train_size": 50000,
+    "eval_size": 10000,
+
+    "num_parallel_calls": 4,
+    "save_summary_steps": 50
+}