Browse Source
        
      
      modify
      
        Signed-off-by: gexy5 <xinyu.ge@zilliz.com>
      
      
        main
      
      
     
    
      
        
          
            
            gexy5
          
          3 years ago
          
         
        
        
       
      
     
    
    
	
		
			
				 1 changed files with 
2 additions and 
0 deletions
			 
			
		 
		
			
				- 
					
					
					 
					tsm.py
				
 
			
		
		
			
			
			
			
			
			
				
				
					
						
							
								
									
	
		
			
				| 
					
					
						
							
						
					
					
				 | 
				@ -95,6 +95,8 @@ class Tsm(NNOperator): | 
			
		
		
	
		
			
				 | 
				 | 
				                    ) | 
				 | 
				 | 
				                    ) | 
			
		
		
	
		
			
				 | 
				 | 
				        inputs = data.to(self.device)[None, ...] | 
				 | 
				 | 
				        inputs = data.to(self.device)[None, ...] | 
			
		
		
	
		
			
				 | 
				 | 
				
 | 
				 | 
				 | 
				
 | 
			
		
		
	
		
			
				 | 
				 | 
				 | 
				 | 
				 | 
				        self.model.eval() | 
			
		
		
	
		
			
				 | 
				 | 
				 | 
				 | 
				 | 
				
 | 
			
		
		
	
		
			
				 | 
				 | 
				        feats = self.model.forward_features(inputs) | 
				 | 
				 | 
				        feats = self.model.forward_features(inputs) | 
			
		
		
	
		
			
				 | 
				 | 
				        if self.model.reshape: | 
				 | 
				 | 
				        if self.model.reshape: | 
			
		
		
	
		
			
				 | 
				 | 
				            if self.model.is_shift and self.model.temporal_pool: | 
				 | 
				 | 
				            if self.model.is_shift and self.model.temporal_pool: | 
			
		
		
	
	
		
			
				| 
					
						
							
						
					
					
					
				 | 
				
  |