diff --git a/classification/main.py b/classification/main.py index 9005c53..1351e22 100644 --- a/classification/main.py +++ b/classification/main.py @@ -34,7 +34,7 @@ from timm.utils import ModelEma as ModelEma from utils.utils_ema import load_checkpoint_ema, load_pretrained_ema, save_checkpoint_ema # print(f"||{torch.multiprocessing.get_start_method()}||", end="") -torch.multiprocessing.set_start_method("spawn", force=True) +# torch.multiprocessing.set_start_method("spawn", force=True) def str2bool(v):