Skip to content

Commit 93ab089

Browse files
committed
TEST: update rPIE dictionary learning tester
1 parent 276b15c commit 93ab089

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/test_multislice_ptycho_rpie_synthesisdictlearn.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
class TestMultislicePtychoRPIESDL(tutils.TungstenDataTester):
1616
@tutils.TungstenDataTester.wrap_recon_tester(name="test_multislice_ptycho_rpie_synthesisdictlearn")
1717
def test_multislice_ptycho_rpie_synthesisdictlearn(self):
18+
self.high_tol = True
1819
self.setup_ptychi(cpu_only=False)
1920

2021
data, probe, pixel_size_m, positions_px = self.load_tungsten_data(additional_opr_modes=3)
@@ -78,7 +79,7 @@ def test_multislice_ptycho_rpie_synthesisdictlearn(self):
7879
options.opr_mode_weight_options.update_relaxation = 1e-2
7980

8081
options.reconstructor_options.batch_size = round(data.shape[0] * 0.1)
81-
options.reconstructor_options.num_epochs = 32
82+
options.reconstructor_options.num_epochs = 8
8283
options.reconstructor_options.allow_nondeterministic_algorithms = False
8384

8485
task = PtychographyTask(options)

0 commit comments

Comments
 (0)