add multidataset

This commit is contained in:
Yifan Yang 2023-06-02 10:34:04 +08:00
parent 771583de0f
commit afc9f9c413

View File

@ -179,6 +179,6 @@ class NewBobScheduler(LRScheduler):
return {
"base_lrs": self.base_lrs,
"prev_metric": self.prev_metric,
"current_metric": current_metric,
"current_metric": self.current_metric,
"current_patient": self.current_patient,
}