diff --git a/README.md b/README.md index 95076d7f1f..928ef1ada4 100644 --- a/README.md +++ b/README.md @@ -135,7 +135,7 @@ trainer = flash.Trainer(max_epochs=1) # 5. Finetune the model trainer.finetune(model, datamodule=datamodule, strategy="freeze") -# 7. Save it! +# 6. Save it! trainer.save_checkpoint("image_classification_model.pt") ``` @@ -359,4 +359,4 @@ We’re excited to continue the strong legacy of opensource software and have be ## License Please observe the Apache 2.0 license that is listed in this repository. In addition -the Lightning framework is Patent Pending. +the Lightning framework is Patent Pending. \ No newline at end of file