results/train.json

{
 "step": 49999,
 "loss": 2.7616,
 "table": 0.8088,
 "value": 5.3021,
 "shape": 0.0011,
 "metric": 4.7414,
 "exact": 0.648,
 "minutes": 117.4,
 "data": "jgalego/bridge2vec-deals",
 "deals": 200000,
 "steps": 50000,
 "batch_size": 1024,
 "learning_rate": 0.0003,
 "value_weight": 0.1,
 "aux_weight": 0.1,
 "metric_weight": 0.3,
 "temperature": 0.1,
 "target_temperature": 0.1,
 "dim": 256,
 "depth": 4,
 "embed_dim": 128,
 "hidden": 1024,
 "parameters": 4912479,
 "runtime_s": 7050,
 "device": "NVIDIA A10G"
}