mirror of https://github.com/InternLM/InternLM
				
				
				
			remove useless code for no-pp
							parent
							
								
									0fa1083780
								
							
						
					
					
						commit
						54e561665e
					
				| 
						 | 
				
			
			@ -202,10 +202,8 @@ class NonPipelineScheduler(BaseScheduler):
 | 
			
		|||
            if return_output_label:
 | 
			
		||||
                outputs.append(_output)
 | 
			
		||||
                labels.append(_label)
 | 
			
		||||
        
 | 
			
		||||
        if not return_output_label:
 | 
			
		||||
            outputs, labels = None, None
 | 
			
		||||
        
 | 
			
		||||
        # Compatible for non-moe
 | 
			
		||||
        if hasattr(gpc.config.model, "num_experts"):
 | 
			
		||||
            return outputs, labels, loss, moe_loss
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue