Change the number of epochs heuristic

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