From 3dc0b75595ac113db0c49ca51a9be33e1ed7e876 Mon Sep 17 00:00:00 2001 From: alanvinx Date: Tue, 29 Oct 2024 19:24:26 +0100 Subject: [PATCH] fix training report, add missing dependency --- environment.yml | 1 + train.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/environment.yml b/environment.yml index 68cd295..5639448 100644 --- a/environment.yml +++ b/environment.yml @@ -15,5 +15,6 @@ dependencies: - pip: - submodules/diff-gaussian-rasterization - submodules/simple-knn + - submodules/fused_ssim - opencv-python - joblib diff --git a/train.py b/train.py index d378d86..8206903 100644 --- a/train.py +++ b/train.py @@ -155,7 +155,7 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoi progress_bar.close() # Log and save - training_report(tb_writer, iteration, Ll1, loss, l1_loss, iter_start.elapsed_time(iter_end), testing_iterations, scene, render, (pipe, background, dataset.train_test_exp, SPARSE_ADAM_AVAILABLE), dataset.train_test_exp) + training_report(tb_writer, iteration, Ll1, loss, l1_loss, iter_start.elapsed_time(iter_end), testing_iterations, scene, render, (pipe, background, 1., SPARSE_ADAM_AVAILABLE, None, dataset.train_test_exp), dataset.train_test_exp) if (iteration in saving_iterations): print("\n[ITER {}] Saving Gaussians".format(iteration)) scene.save(iteration)