| { | |
| "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" | |
| } |