fix missing self. (#61)

Co-authored-by: 许文卿 <xuwenqing@XWQ.local>
This commit is contained in:
Xu Wenqing
2023-01-05 17:44:31 +08:00
committed by GitHub
parent 1d0a8ea491
commit 705bf67283

View File

@@ -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),