We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 015fbe4 commit c3992d5Copy full SHA for c3992d5
validate.py
@@ -389,7 +389,7 @@ def _try_run(args, initial_batch_size):
389
try:
390
if 'cuda' in args.device and torch.cuda.is_available():
391
torch.cuda.empty_cache()
392
- elif "npu" in args.device and torch.npu.is_available():
+ elif "npu" in args.device and torch.npu.is_available():
393
torch.npu.empty_cache()
394
results = validate(args)
395
return results
0 commit comments