benchmark script

nerfstudio-project · Aug 8, 2024 · 9e33225 · 9e33225
1 parent d313a62
commit 9e33225
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/examples/benchmark.py b/examples/benchmark.py
@@ -11,7 +11,7 @@
 
 @dataclass
 class BenchmarkConfig:
-    """Benchmark config"""
+    """Baseline benchmark config"""
 
     # trainer to run
     trainer: str = "simple_trainer.py"
@@ -39,7 +39,7 @@ class BenchmarkConfig:
     model_configs: dict = field(default_factory=dict)
 
 
-# Configurations to run
+# Configurations of different GSPLAT options
 baseline_config = BenchmarkConfig()
 absgrad_config = BenchmarkConfig(
     result_dir="results/absgrad",
@@ -53,11 +53,12 @@ class BenchmarkConfig:
     result_dir="results/mcmc",
 )
 
+# Configs to run
 configs_to_run = [
     baseline_config,
-    # mcmc_config,
-    # absgrad_config,
-    # antialiased_config,
+    mcmc_config,
+    absgrad_config,
+    antialiased_config,
 ]