model_checkpoint_path: "model_weights"
all_model_checkpoint_paths: "model_weights"