@nareshr8, you can reduce n_samples to limit the number of perturbed examples and n_steps to reduce the number of integral approximation steps.
With our recent improvements (the PR got merged a couple of days ago) you should be able to adjust internal_batch_size and run IG for very large number of steps.

  
      github.com/pytorch/captum
  
  
    
  
    
  

  
    
      Internal Batching Improvements (#333) 34
    

    
      
        committed 
        
          
        
        Apr 1, 2020
      
      

      
        
          
          vivekmig
        
        
      

      
        
          +337
          -85
        
      
    

  



  
    Summary:
This PR improves internal batching to avoid creating an expanded tensor initially and simply recursively compute each subset of steps and...