ViT_2D.ckpt #1