Add train results

results/train.json

@@ -0,0 +1,19 @@
+{
+ "step": 4,
+ "loss": 2.9403,
+ "contrastive": 2.5824,
+ "invariants": 3.5794,
+ "pair_top1": 0.125,
+ "minutes": 0.1,
+ "data": "jgalego/knot2vec-diagrams",
+ "steps": 5,
+ "batch_size": 16,
+ "learning_rate": 0.0003,
+ "temperature": 0.05,
+ "aux_weight": 0.1,
+ "dim": 64,
+ "depth": 2,
+ "parameters": 134479,
+ "runtime_s": 8,
+ "device": "cpu"
+}