Skip to content

Commit ebf5921

Browse files
committed
RAM doesn't need an artifact removal
1 parent 11532b0 commit ebf5921

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

  • deepinv_bench/benchmarks/cbsd500_gaussian_denoising/solvers

deepinv_bench/benchmarks/cbsd500_gaussian_denoising/solvers/ram.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,7 @@ def set_objective(self, train_dataset=None, physics=None):
1313
device = (
1414
dinv.utils.get_freer_gpu() if torch.cuda.is_available() else "cpu"
1515
)
16-
self.model = dinv.models.ArtifactRemoval(
17-
dinv.models.RAM(), device=device
18-
)
16+
self.model = dinv.models.RAM()
1917
self.model.device = device
2018

2119
def run(self, _):

0 commit comments

Comments
 (0)