fix: wrap classifier batch inference in torch.no_grad() to stop memory leak - #662

fix: wrap classifier batch inference in torch.no_grad() to stop memory leak#662

-Step- (StepFPV) wants to merge 1 commit into

microsoft:mainmicrosoft/Biodiversity:mainfrom

StepFPV:fix/timm-classifier-memory-leakStepFPV/Biodiversity:fix/timm-classifier-memory-leakCopy head branch name to clipboard