X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=speed.py;h=075b07ee3bb54deda3dc34b02cc05074d4d39381;hb=5db6798d929c15e1517ec10c1a9211f870ec977e;hp=e5b0e3a6d13c777afbe6a58bc9cacd46e35293ff;hpb=20285925e51c7adc6e7bb64bb9d2a5cab92c6aac;p=pytorch.git

diff --git a/speed.py b/speed.py
index e5b0e3a..075b07e 100755
--- 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}')