diff options
| author | Daniil Kazantsev <dkazanc@hotmail.com> | 2019-12-10 10:40:58 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-12-10 10:40:58 +0000 |
| commit | a19c2ef5e1d8ebfdfe97d6755cd3d7d753e73541 (patch) | |
| tree | b84242f17b05ee5251a85dd911d4e9c87f3a8e2e /test | |
| parent | d1585fb280ead79b2bf3962c3e6d492b71acb723 (diff) | |
| parent | 041bdac2cda4633906600d8c77c37245b1e2e817 (diff) | |
| download | regularization-a19c2ef5e1d8ebfdfe97d6755cd3d7d753e73541.tar.gz regularization-a19c2ef5e1d8ebfdfe97d6755cd3d7d753e73541.tar.bz2 regularization-a19c2ef5e1d8ebfdfe97d6755cd3d7d753e73541.tar.xz regularization-a19c2ef5e1d8ebfdfe97d6755cd3d7d753e73541.zip | |
Merge pull request #138 from vais-ral/pdtv_gpu_fix
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.")
|