infer_part = partial(infer, out_dir_path=cfg.EXP_DIR,model=model_copy.to('cpu'))
multiprocessing.Process(target=infer_part).start()