diff --git a/mu_map/random_search/cgan.py b/mu_map/random_search/cgan.py index 8ac6071788b4d91292d273d8df9801e12f36bc65..b6984d3cff0020fede66cbb173e352e242433b80 100644 --- a/mu_map/random_search/cgan.py +++ b/mu_map/random_search/cgan.py @@ -67,7 +67,7 @@ class cGANRandomSearch(RandomSearch): self.param_sampler["patch_offset"] = ChoiceSampler([0]) self.param_sampler["patch_size"] = ChoiceSampler([32, 64]) self.param_sampler["patch_number"] = DependentChoiceSampler(patch_number) - self.param_sampler["scatter_correction"] = ChoiceSampler([False]) + self.param_sampler["scatter_correction"] = ChoiceSampler([False, True]) self.param_sampler["shuffle"] = ChoiceSampler([False, True]) self.param_sampler["normalization"] = ChoiceSampler( [MeanNormTransform(), MaxNormTransform(), GaussianNormTransform()]