diff options
| author | dkazanc <dkazanc@hotmail.com> | 2019-12-09 14:08:21 +0000 |
|---|---|---|
| committer | dkazanc <dkazanc@hotmail.com> | 2019-12-09 14:08:21 +0000 |
| commit | 970e82c7ee5fedab771480f42293963fdc32d17b (patch) | |
| tree | a258214525adf239ecdc76952cc69fd4de801d16 /test | |
| parent | d1585fb280ead79b2bf3962c3e6d492b71acb723 (diff) | |
| download | regularization-970e82c7ee5fedab771480f42293963fdc32d17b.tar.gz regularization-970e82c7ee5fedab771480f42293963fdc32d17b.tar.bz2 regularization-970e82c7ee5fedab771480f42293963fdc32d17b.tar.xz regularization-970e82c7ee5fedab771480f42293963fdc32d17b.zip | |
fixes gpu issues with pdtv
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_CPU_regularisers.py | 2 | ||||
| -rwxr-xr-x | test/test_run_test.py | 9 |
2 files changed, 4 insertions, 7 deletions
diff --git a/test/test_CPU_regularisers.py b/test/test_CPU_regularisers.py index 266ca8a..1eba479 100644 --- a/test/test_CPU_regularisers.py +++ b/test/test_CPU_regularisers.py @@ -42,7 +42,7 @@ class TestRegularisers(unittest.TestCase): def test_PD_TV_CPU(self): Im,input,ref = self.getPars() - pd_cpu,info = PD_TV(input, 0.02, 300, 0.0, 0, 0, 8, 0.0025, 'cpu'); + pd_cpu,info = PD_TV(input, 0.02, 300, 0.0, 0, 0, 8, 'cpu'); rms = rmse(Im, pd_cpu) diff --git a/test/test_run_test.py b/test/test_run_test.py index 1707aec..f562593 100755 --- a/test/test_run_test.py +++ b/test/test_run_test.py @@ -200,8 +200,7 @@ class TestRegularisers(unittest.TestCase): 'tolerance_constant':0.0,\
'methodTV': 0 ,\
'nonneg': 0,
- 'lipschitz_const' : 8,
- 'tau' : 0.0025}
+ 'lipschitz_const' : 8}
print ("#############PD TV CPU####################")
start_time = timeit.default_timer()
@@ -211,8 +210,7 @@ class TestRegularisers(unittest.TestCase): pars['tolerance_constant'],
pars['methodTV'],
pars['nonneg'],
- pars['lipschitz_const'],
- pars['tau'],'cpu')
+ pars['lipschitz_const'],'cpu')
rms = rmse(Im, pd_cpu)
pars['rmse'] = rms
@@ -230,8 +228,7 @@ class TestRegularisers(unittest.TestCase): pars['tolerance_constant'],
pars['methodTV'],
pars['nonneg'],
- pars['lipschitz_const'],
- pars['tau'],'gpu')
+ pars['lipschitz_const'],'gpu')
except ValueError as ve:
self.skipTest("Results not comparable. GPU computing error.")
|