would be possible to extract features from the saved model and save them as a .np file?
I have saved the model using
     if epoch % 10 == 9:
             
            torch.save({
                    'epoch': epoch + 1,
                    'state_dict': model.state_dict(),
                    'optimizer': optimizer.state_dict(),
            }, 'NAS_exp_epoch{}.pth'.format(epoch+1))
             

I wonder how can I extract features from the model for each image patch?