Bladeren bron

Change the number of epochs heuristic

Vadim Markovtsev 7 jaren geleden
bovenliggende
commit
703e28fb35
1 gewijzigde bestanden met toevoegingen van 3 en 3 verwijderingen
  1. 3 3
      labours.py

+ 3 - 3
labours.py

@@ -676,13 +676,13 @@ def train_embeddings(index, matrix, tmpdir, shard_size=4096):
         swivel.FLAGS.submatrix_cols = shard_size
         if len(meta_index) < 10000:
             embedding_size = 50
-            num_epochs = 200
+            num_epochs = 500
         elif len(meta_index) < 100000:
             embedding_size = 100
-            num_epochs = 250
+            num_epochs = 300
         elif len(meta_index) < 500000:
             embedding_size = 200
-            num_epochs = 300
+            num_epochs = 250
         else:
             embedding_size = 300
             num_epochs = 200