{
 "step": 9999,
 "loss": 1.1689,
 "contrastive": 0.5045,
 "invariants": 0.6644,
 "pair_top1": 0.869,
 "minutes": 92.2,
 "data": "jgalego/knot2vec-diagrams-hard",
 "steps": 10000,
 "batch_size": 512,
 "learning_rate": 0.0001,
 "temperature": 0.05,
 "aux_weight": 1.0,
 "init": "jgalego/knot2vec",
 "dim": 256,
 "depth": 6,
 "parameters": 4974095,
 "runtime_s": 5533,
 "device": "NVIDIA A10G"
}