commit

a0405u · Dec 29, 2020 · e163ef3 · e163ef3
1 parent 4f447fa
commit e163ef3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lab3/lab3.py b/lab3/lab3.py
@@ -93,7 +93,7 @@ def predict(image):
     #output = model_trt(input)
     output = model(input)
 
-    probability torch.nn.functional.softmax(output, dim = 1)
+    probability = torch.nn.functional.softmax(output, dim = 1)
 
     print("image processed in {}s".format(round(time.time() - timest, 3)))