Skip to content

Commit 3dc0b75

Browse files
committed
fix training report, add missing dependency
1 parent 9e1c2c6 commit 3dc0b75

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

environment.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,6 @@ dependencies:
1515
- pip:
1616
- submodules/diff-gaussian-rasterization
1717
- submodules/simple-knn
18+
- submodules/fused_ssim
1819
- opencv-python
1920
- joblib

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoi
155155
progress_bar.close()
156156

157157
# Log and save
158-
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)
158+
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)
159159
if (iteration in saving_iterations):
160160
print("\n[ITER {}] Saving Gaussians".format(iteration))
161161
scene.save(iteration)

0 commit comments

Comments
 (0)