fix run issue

2026-02-03 15:01:30 +08:00
parent e556f17ebc
commit feb69b7a55
3 changed files with 216 additions and 3 deletions
--- a/trainers/maple.py
+++ b/trainers/maple.py
@@ -19,7 +19,7 @@ from dassl.metrics import compute_accuracy
 from dassl.utils import load_pretrained_weights, load_checkpoint, mkdir_if_missing
 from dassl.optim import build_optimizer, build_lr_scheduler
 from dassl.evaluation import Classification,EvaluatorBase
-from pygrad.pcgrad import PCGrad
+#from pygrad.pcgrad import PCGrad
 from datasets.data_manager import DataManager
 from dassl.data.datasets import build_dataset