Free cached memory inside the batch-size probe, not after estimation#596

CompRhys merged 1 commit into

TorchSim:mainTorchSim/torch-sim:mainfrom

avnoordenne:fix/empty-cache-after-batch-size-probeavnoordenne/torch-sim:fix/empty-cache-after-batch-size-probeCopy head branch name to clipboard