feat: train fixed-size classifiers and export class probabilities

2026-10-14 17:08:14 +08:00
parent 63bab36147
commit f9a4032783
11 changed files with 478 additions and 208 deletions
@@ -0,0 +1,8 @@
+_base_: film_tiles128_resnet18.yaml
+work_dir: runs/classify_tiles128_mobilenet_v3_small
+model:
+  backbone:
+    arch: mobilenet_v3_small
+train:
+  swanlab:
+    tags: [classify, mobilenet_v3_small, tiles128]