# TODO: Save the checkpoint neural_net.save_model_checkpoint('checkpoint_1.pth', datasets['train'].class_to_idx)
