From 5ee7d04e8a0ba8fcf42f61c63ffaaad1b25b89d2 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sat, 5 Sep 2020 10:01:54 +0200 Subject: [PATCH] Update. --- speed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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}') -- 2.20.1