Thanks, I could not use this solution, because the model is previously saved by someone else and I just want to test it. However, I found the solution by changing the model to the following:
self.model = self.model.module.cpu()
More info can be found in the following issue:
  
      github.com/JDAI-CV/FaceX-Zoo
  
  
    
  
	  
  

  
    
      Test face_pipline.py on CPU
    

    
      
        opened 
        
          
        
        Feb 13, 2021
      
      

        
          closed 
        
          
        
        Feb 14, 2021
      
        

      
        
          
          marziehoghbaie
        
      
    
  



  Hi,
I like to test face_pipline.py on cpu, but I keep taking the following error:
RuntimeError: module must have its parameters and buffers...