results/train.json

{
 "step": 19999,
 "loss": 0.4862,
 "contrastive": 0.2473,
 "heads": 0.7963,
 "pair_top1": 0.902,
 "minutes": 16.8,
 "data": "jgalego/orbit2vec-windows",
 "pairs": 302056,
 "objects": 45261,
 "steps": 20000,
 "batch_size": 512,
 "learning_rate": 0.0003,
 "temperature": 0.05,
 "aux_weight": 0.3,
 "dim": 256,
 "depth": 4,
 "parameters": 3314479,
 "runtime_s": 1009,
 "device": "NVIDIA A10G"
}