From a5864af1dc58fc7372022ae2e4c6227c30f9b46f Mon Sep 17 00:00:00 2001 From: UnknownObject Date: Wed, 6 Dec 2023 19:53:41 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9C=80=E7=BB=88=E7=89=88=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ver3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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():