Skip to content

GitLab

Commit 224ba1a7 authored by Ruilong Li's avatar Ruilong Li
Browse files

update readme

parent c3ab153e
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
README.md
View file @ 224ba1a7
...@@ -12,9 +12,9 @@ python examples/trainval.py ...@@ -12,9 +12,9 @@ python examples/trainval.py
| trainval (35k, 1<<16) | Lego | Mic | Materials ｜ | trainval (35k, 1<<16) | Lego | Mic | Materials ｜
| - | - | - | - ｜ | - | - | - | - ｜
| Time | 325s | 357s ｜ 354s | | Time | 325s | 357s | 360s |
| PSNR | 36.20 | 36.55 ｜ 29.63 | | PSNR | 36.20 | 36.55 | 29.59 |
| FPS | 12.56 | 25.54 | | FPS | 12.56 | 25.54 | 9.65 |
Tested with the default settings on the Lego test set. Tested with the default settings on the Lego test set.
......
examples/trainval.py
View file @ 224ba1a7
...@@ -90,7 +90,7 @@ if __name__ == "__main__": ...@@ -90,7 +90,7 @@ if __name__ == "__main__":
torch.manual_seed(42) torch.manual_seed(42)
device = "cuda:0" device = "cuda:0"
scene = "materials" scene = "lego"
# setup dataset # setup dataset
train_dataset = SubjectLoader( train_dataset = SubjectLoader(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment