Warning triggered whenever I'm using the forward pass the second time

[ormistrix]

Hi,

I'm getting this warning when i'm calling to farward function of ---- the second time

""" /usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py:1194: UserWarning: operator() profile_node %385 : int[] = prim::profile_ivalue(%383) does not have profile information (Triggered internally at ../torch/csrc/jit/codegen/cuda/graph_fuser.cpp:105.) return forward_call(*input, **kwargs) """

Here is the code i'm using

    # INIT
    self.face_detector = facer.face_detector('retinaface/mobilenet', device=device)
    conf_name = 'celebm/448'
    model_path = 'https://github.com/FacePerceiver/facer/releases/download/models-v1/face_parsing.farl.celebm.main_ema_181500_jit.pt'
    self.face_parser_for_source = FaRLFaceParser(conf_name, model_path=model_path, device=device).to(device)        
    
    # HERE IS THE CODE THAT TRIGER THE WARNING THE SECOND TIME I'M CALL THE FUNC
     with torch.inference_mode():
                faces = self.face_detector(image)
                faces = self.face_parser_for_source(image, faces)

BTW If I'm init the function before I'm calling it the second time there are no warnings, though it's twice the running time )=

Thanks for your help

[icech]

I also encountered this promblem! It runs so slowly in the first time.