From 31d2902b1644d76be7acf48bddb866e75665a4cf Mon Sep 17 00:00:00 2001 From: A0405u Date: Tue, 29 Dec 2020 16:38:55 +0300 Subject: [PATCH] commit --- lab3/lab3.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lab3/lab3.py b/lab3/lab3.py index d25687d..c884557 100644 --- a/lab3/lab3.py +++ b/lab3/lab3.py @@ -93,7 +93,8 @@ def predict(image): #output = model_trt(input) output = model(input) - probability = torch.nn.Softmax(output) + sm = torch.nn.Softmax() + probability = sm(output) print("image processed in {}s".format(round(time.time() - timest, 3)))