最终版本

2 years ago · 8f964a02a5
parent a5864af1dc
commit 8f964a02a5
1 changed files with 1 additions and 1 deletions
--- a/ver3.py
+++ b/ver3.py
@ -48,7 +48,7 @@ def train():

    model = vision_learner(dls, resnet34, metrics=error_rate)
    model.fine_tune(5, freeze_epochs=3) #5 - 训练的轮次， 3 - 冻结的轮次
-    model.export('G:\\Users\\15819\\Desktop\\model01.pkl')
+    model.export('G:\\Users\\15819\\Desktop\\model01.pkl') # M20.02.pkl

 def main():
    train()