Update.

[pytorch.git] / speed.py

diff --git a/speed.py b/speed.py
index e5b0e3a..075b07e 100755 (executable)
--- a/speed.py
+++ b/speed.py
@@ -42,4 +42,4 @@ for t in [ torch.float32, torch.float16 ]:
 
     except:
 
-        print(f'Cannot try with {t}')
+        print(f'{t} is not available on {device}')