diff --git a/README.md b/README.md index 8afd823c..783fd342 100644 --- a/README.md +++ b/README.md @@ -55,7 +55,7 @@ trainer = Trainer( lr_scheduler_milestones=None ) -# init model +# init model and train model = My_Model() trainer.fit(model) ```