Add train results

results/train.json

@@ -1,19 +1,19 @@
 {
- "step": 4,
- "loss": 2.9403,
- "contrastive": 2.5824,
- "invariants": 3.5794,
- "pair_top1": 0.125,
- "minutes": 0.1,
+ "step": 19999,
+ "loss": 0.2043,
+ "contrastive": 0.1145,
+ "invariants": 0.8985,
+ "pair_top1": 0.971,
+ "minutes": 101.7,
  "data": "jgalego/knot2vec-diagrams",
- "steps": 5,
- "batch_size": 16,
+ "steps": 20000,
+ "batch_size": 512,
  "learning_rate": 0.0003,
  "temperature": 0.05,
  "aux_weight": 0.1,
- "dim": 64,
- "depth": 2,
- "parameters": 134479,
- "runtime_s": 8,
- "device": "cpu"
+ "dim": 256,
+ "depth": 6,
+ "parameters": 4924943,
+ "runtime_s": 6106,
+ "device": "NVIDIA A10G"
 }