Mtrini-SVL-2.0 / training_log.json
CompiwerAI's picture
Mtrini-SVL-2.0 round 7, Gated Self-Training with DPO (5 steps)
fe356f5 verified
Raw History Blame Contribute Delete
1.66 kB
{"steps": 5, "max_steps": 5, "log": [{"loss": 0.6931473016738892, "grad_norm": 17.957149505615234, "learning_rate": 1.6666666666666667e-06, "entropy": 0.6042060684412718, "num_tokens": 129113.0, "logits/chosen": -2.031570067427444, "logits/rejected": -2.066084378741378, "mean_token_accuracy": 0.8573183789849281, "rewards/chosen": 0.0, "rewards/rejected": 0.0, "rewards/accuracies": 0.0, "rewards/margins": 0.0, "logps/chosen": -1720.7409473061562, "logps/rejected": -1911.8703248500824, "epoch": 0.43243243243243246, "step": 2}, {"loss": 0.7156742215156555, "grad_norm": 14.256765365600586, "learning_rate": 5e-06, "entropy": 0.7723287399858236, "num_tokens": 235146.0, "logits/chosen": -1.9302834164158855, "logits/rejected": -1.978542090787214, "mean_token_accuracy": 0.832091361284256, "rewards/chosen": -0.020962871101801284, "rewards/rejected": 0.020634264779801015, "rewards/accuracies": 0.4375, "rewards/margins": -0.04159713693661615, "logps/chosen": -1395.894810795784, "logps/rejected": -1786.6366989016533, "epoch": 0.8648648648648649, "step": 4}, {"train_runtime": 249.6526, "train_samples_per_second": 0.16, "train_steps_per_second": 0.02, "total_flos": 1.5657738870199104e+16, "train_loss": 0.7078817367553711, "entropy": 0.6014999866485595, "num_tokens": 277106.0, "logits/chosen": -2.0039181554060184, "logits/rejected": -1.9878743479109093, "mean_token_accuracy": 0.8745649218559265, "rewards/chosen": -0.044874840695410964, "rewards/rejected": -0.00777633311226964, "rewards/accuracies": 0.6, "rewards/margins": -0.03709850497543812, "logps/chosen": -1500.4318752288818, "logps/rejected": -1871.2910789489747, "epoch": 1.0, "step": 5}]}