config.py
Signed-off-by: Andrii Skliar <askliar@nvidia.com> Co-authored-by: Andrii Skliar <askliar@nvidia.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache