mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-25 12:39:25 +01:00
@@ -423,7 +423,7 @@ class ImageClassifitionTrainer(BaseTrainer):
|
||||
classname_path = osp.join(data_root, 'classname.txt')
|
||||
classes = classname_path if osp.exists(classname_path) else None
|
||||
else:
|
||||
classes = cfg.dataset.classes
|
||||
classes = self.cfg.dataset.classes
|
||||
dataset = MmDataset(
|
||||
self.eval_dataset,
|
||||
pipeline=preprocess_transform(self.cfg.preprocessor.val),
|
||||
|
||||
Reference in New Issue
Block a user