train

2026-08-24 09:40:31 +08:00
parent 5199e3040f
commit 2bf7db6bb9
23 changed files with 51 additions and 105 deletions
@@ -26,22 +26,25 @@ if mode == "toy":
    ]
    num_classes = len(label_name)
 elif mode == "benchmark":
-    epoch = 100  # convnext->100, resnet18->50
-    lr = 1e-4
+    epoch = 50  # resnet=80, resnet18=50, convnext=100
+    lr = 1e-4  # resnet, resnet18=1e-4
+    weight_decay = 1e-4  # resnet=1e-3, resnet18=1e-4
    batch_size = 8
    input_size = 224

    # 分类
    label_name = [
-        "american_shorthair",
-        "bengal",
-        "british_shorthair",
-        "exotic_shorthair",
-        "maine_coon",
-        "ragdoll",
-        "scottish_fold",
-        "siamese",
-        "sphynx",
-        "turkish_van",
+        "american_shorthair",  # 美国短毛猫
+        "british_shorthair",  # 英国短毛猫
+        "ragdoll",  # 布偶猫
+        "exotic_shorthair",  # 异国短毛猫
+        "maine_coon",  # 缅因猫
+        "siamese",  # 暹罗猫
+        "sphynx",  # 斯芬克斯猫
+        "turkish_van",  # 土耳其梵猫
+        "bengal",  # 孟加拉豹猫
+        "scottish_fold",  # 苏格兰折耳猫
+        "none",  # 风景人物
+        "other",  # 其他动物
    ]
    num_classes = len(label_name)