最终版本

2023-12-06 19:55:42 +08:00
parent 257fb99811
commit a5864af1dc
1 changed files with 1 additions and 1 deletions
@@ -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():