From 9e338578f8df62935c0d656b30cdac55fd49dc9e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thomas=20M=C3=BCller?= <thomas94@gmx.net>
Date: Tue, 17 Jan 2023 09:50:45 +0100
Subject: [PATCH] Make run.py and main.cu snapshot flags compatible
---
scripts/run.py | 2 +-
src/main.cu | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/scripts/run.py b/scripts/run.py
index 3f576e9..cbcf4a3 100644
--- a/scripts/run.py
+++ b/scripts/run.py
@@ -33,7 +33,7 @@ def parse_args():
parser.add_argument("--mode", default="", type=str, help=argparse.SUPPRESS) # deprecated
parser.add_argument("--network", default="", help="Path to the network config. Uses the scene's default if unspecified.")
- parser.add_argument("--load_snapshot", default="", help="Load this snapshot before training. recommended extension: .msgpack")
+ parser.add_argument("--load_snapshot", "--snapshot", default="", help="Load this snapshot before training. recommended extension: .msgpack")
parser.add_argument("--save_snapshot", default="", help="Save this snapshot after training. recommended extension: .msgpack")
parser.add_argument("--nerf_compatibility", action="store_true", help="Matches parameters with original NeRF. Can cause slowness and worse results on some scenes.")
diff --git a/src/main.cu b/src/main.cu
index 95628b7..078014f 100644
--- a/src/main.cu
+++ b/src/main.cu
@@ -79,7 +79,7 @@ int main(int argc, char** argv) {
parser,
"SNAPSHOT",
"Optional snapshot to load upon startup.",
- {"snapshot"},
+ {"snapshot", "load_snapshot"},
};
ValueFlag<uint32_t> width_flag{
--
GitLab