diff --git a/ver3.py b/ver3.py
index 5ce18dd..742090d 100644
--- a/ver3.py
+++ b/ver3.py
@@ -47,7 +47,7 @@ def train():
     ).dataloaders(data_path, num_workers=4, bs=batch_size)
 
     model = vision_learner(dls, resnet34, metrics=error_rate)
-    model.fine_tune(5, freeze_epochs=3)
+    model.fine_tune(5, freeze_epochs=3) #5 - 训练的轮次， 3 - 冻结的轮次
     model.export('G:\\Users\\15819\\Desktop\\model01.pkl')
 
 def main():