Nevermind, found the answer here!

  
      github.com/pytorch/pytorch
  
  
    
  
	  
  

  
    
      Get a single batch from DataLoader without iterating 2.4k
    

    
      
        opened 
        
          
        
        Jun 26, 2017
      
      

        
          closed 
        
          
        
        Jun 26, 2017
      
        

      
        
          
          narendasan
        
      
    
  



  Is it possible to get a single batch from a DataLoader? Currently, I setup a for loop and return a batch...