Rain-Bus/DAPT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix run issue

Browse Source
This commit is contained in:
Rain&Bus
2026-02-03 15:01:30 +08:00
parent e556f17ebc
commit feb69b7a55
3 changed files with 216 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@@ -291,7 +291,7 @@ def main(args):
trainer.test_withlabel()
# trainer.test_multi_label() Evaluation for VOC12
import shutil
shutil.rmtree(original_weight_output+'/MultiModalPromptLearner')
# shutil.rmtree(original_weight_output+'/MultiModalPromptLearner')
else:
original_weight_output = 'output/' + '/'.join(trainer.output_dir.split('/')[1:])
trainer.load_model(original_weight_output)
@@ -304,7 +304,7 @@ def main(args):
trainer.train()
if cfg.DATASET.SUBSAMPLE_CLASSES != 'base':
import shutil
shutil.rmtree(trainer.output_dir+'/MultiModalPromptLearner')
# shutil.rmtree(trainer.output_dir+'/MultiModalPromptLearner')
if __name__ == "__main__":