增加学习率调度方式

2025-06-13 10:45:53 +08:00
parent 37ecef40f7
commit 1803f319a5
13 changed files with 319 additions and 294 deletions
--- a/tools/write_feature_json.py
+++ b/tools/write_feature_json.py
@ -50,7 +50,7 @@ class FeatureExtractor:
                raise FileNotFoundError(f"Model weights file not found: {model_path}")

            # Initialize model
-            model = resnet18().to(self.conf['base']['device'])
+            model = resnet18(scale=self.conf['models']['channel_ratio']).to(self.conf['base']['device'])

            # Handle multi-GPU case
            if conf['base']['distributed']: