Skip to content

Commit ee0f172

Browse files
author
chengshen
committed
update REAME
1 parent d852d70 commit ee0f172

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

README.md

+8
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@ For DnD benchmark, we use MixUp additionally:
3838
python train_mge.py -d prepared_data_path -n num_gpus --dnd
3939
```
4040

41+
### Begin testing:
42+
Download the pretrained checkpoint and use:
43+
44+
```
45+
python test.py -d prepared_data_path -c checkpoint_path
46+
```
47+
The result is **PSNR 39.765**.
48+
4149

4250

4351
## Pretrained model

train_mge.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
def main():
2525
parser = argparse.ArgumentParser(description="MegEngine NBNet")
26-
parser.add_argument("-d", "--data", default="/data/sidd", metavar="DIR", help="path to imagenet dataset")
26+
parser.add_argument("-d", "--data", default="/data/sidd", metavar="DIR", help="path to sidd dataset")
2727
parser.add_argument("--dnd", action='store_true', help="training for dnd benchmark")
2828
parser.add_argument(
2929
"-a",

0 commit comments

Comments
 (0)