From 970e82c7ee5fedab771480f42293963fdc32d17b Mon Sep 17 00:00:00 2001
From: dkazanc <dkazanc@hotmail.com>
Date: Mon, 9 Dec 2019 14:08:21 +0000
Subject: fixes gpu issues with pdtv
---
src/Python/ccpi/filters/regularisers.py | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
(limited to 'src/Python/ccpi')
diff --git a/src/Python/ccpi/filters/regularisers.py b/src/Python/ccpi/filters/regularisers.py
index 5f4001a..e3a984e 100644
--- a/src/Python/ccpi/filters/regularisers.py
+++ b/src/Python/ccpi/filters/regularisers.py
@@ -53,7 +53,7 @@ def FGP_TV(inputData, regularisation_parameter,iterations,
.format(device))
def PD_TV(inputData, regularisation_parameter, iterations,
- tolerance_param, methodTV, nonneg, lipschitz_const, tau, device='cpu'):
+ tolerance_param, methodTV, nonneg, lipschitz_const, device='cpu'):
if device == 'cpu':
return TV_PD_CPU(inputData,
regularisation_parameter,
@@ -61,8 +61,7 @@ def PD_TV(inputData, regularisation_parameter, iterations,
tolerance_param,
methodTV,
nonneg,
- lipschitz_const,
- tau)
+ lipschitz_const)
elif device == 'gpu' and gpu_enabled:
return TV_PD_GPU(inputData,
regularisation_parameter,
@@ -70,8 +69,7 @@ def PD_TV(inputData, regularisation_parameter, iterations,
tolerance_param,
methodTV,
nonneg,
- lipschitz_const,
- tau)
+ lipschitz_const)
else:
if not gpu_enabled and device == 'gpu':
raise ValueError ('GPU is not available')
--
cgit v1.2.3