wanted to update that the problem was solved in the attached link, incase someone will encounter the same issue.
  
      github.com/pytorch/vision
  
  
    
  
	  
  

  
    
      Using torchvision roi_align in libtorch c++ jit modules 26
    

    
      
        opened 
        
          
        
        Jan 26, 2021
      
      

        
          closed 
        
          
        
        Feb 7, 2021
      
        

      
        
          
          natangold85
        
      
    
  



  🐛 Bug
Hi, I’m trying to use libtorch 1.7.1 to load a jit model that is created with pytorch 1.5.1 and torchvision...



    module: c++ frontend
    module: ops
    question
    topic: object detection