diff --git a/ver3.py b/ver3.py index 742090d..0516c3c 100644 --- 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()