clarifty pretrain model name

Zeqiang-Lai · Jul 24, 2022 · 836a727 · 836a727
1 parent 0c37a92
commit 836a727
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -37,6 +37,7 @@ pip install -e .
 3. **Run cli or playgrounds**
 
 - Download the [sample input](https://1drv.ms/u/s!AomvdxwcLmYImEc-Yfj2B2FBGEb0?e=vShXe9) if you don't have one.
+- Download the [pretrained model](https://1drv.ms/u/s!AomvdxwcLmYImDiK2NiIkn_LjsOa?e=PPfex2), `grunet.pth`, put it in `playgrounds/grunet.pth`.
 
 ```shell
 # run cli

diff --git a/cli/task.py b/cli/task.py
@@ -87,7 +87,7 @@ def misr(gt, device, cfg):
         'output_path': 'Lehavim_0910-1717_deblur',
         'denoiser': {
             'type': 'grunet',
-            'model_path': 'playgrounds/unet_qrnn3d.pth',
+            'model_path': 'playgrounds/grunet.pth',
         },
         'solver': 'admm',
         'device': 'cuda',