Signed-off-by: Kunshang Ji <jikunshang95@gmail.com> Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache
mypy