2000 steps checkpoint: https://wandb.ai/open-assistant/supervised-finetuning/runs/w43zluqz