Hi, @promach if you need any help with inference code, you can check the example of my code @

  
      
      GitHub
  
  
    

amrit-das/Custom-Model-Training-PyTorch 15

PyTorch Framework based API for Custom Model Training - amrit-das/Custom-Model-Training-PyTorch


  
  
    
    
  
  


You’ll find a code named inference.py
batch and label are two variables that has been used to extrapolate data from the test_loaders
Labels has the class_name and batch probably has the images