Remove extra space

device_amp_cleanup
Ross Wightman 2024-10-18 14:54:16 -07:00
parent 015fbe457a
commit c3992d5c4c
1 changed files with 1 additions and 1 deletions

View File

@ -389,7 +389,7 @@ def _try_run(args, initial_batch_size):
try:
if 'cuda' in args.device and torch.cuda.is_available():
torch.cuda.empty_cache()
elif "npu" in args.device and torch.npu.is_available():
elif "npu" in args.device and torch.npu.is_available():
torch.npu.empty_cache()
results = validate(args)
return results