PFEemp2024's picture
solving GPU error for previous version
4a1df2e
from anonymous_demo.functional.dataset.dataset_manager import detect_infer_dataset