fixed run_time type

MartinXPN · Feb 3, 2021 · c4c7575 · c4c7575
1 parent 67aa6c7
commit c4c7575
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -39,7 +39,7 @@ def predict(model_path: Union[str, IO],
 
     res = model.validation_step(graph, batch_idx=0)[0]
     end = time.time()
-    res['run_time'] = end - start,
+    res['run_time'] = end - start
     pprint(res, sort_dicts=False)