From a0090e4cad3ce3a5e52ad4239e5f5e2f9c70ebff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thomas=20M=C3=BCller?= <thomas94@gmx.net>
Date: Sun, 15 Jan 2023 07:08:26 +0100
Subject: [PATCH] NeRF: fix broken training on some scenes
---
src/testbed_nerf.cu | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/testbed_nerf.cu b/src/testbed_nerf.cu
index cc054f8..0f9f31e 100644
--- a/src/testbed_nerf.cu
+++ b/src/testbed_nerf.cu
@@ -3057,7 +3057,7 @@ void Testbed::train_nerf(uint32_t target_batch_size, bool get_loss_scalar, cudaS
void Testbed::train_nerf_step(uint32_t target_batch_size, Testbed::NerfCounters& counters, cudaStream_t stream) {
const uint32_t padded_output_width = m_network->padded_output_width();
- const uint32_t max_samples = target_batch_size * 4; // Somewhat of a worst case
+ const uint32_t max_samples = target_batch_size * 16; // Somewhat of a worst case
const uint32_t floats_per_coord = sizeof(NerfCoordinate) / sizeof(float) + m_nerf_network->n_extra_dims();
const uint32_t extra_stride = m_nerf_network->n_extra_dims() * sizeof(float); // extra stride on top of base NerfCoordinate struct
--
GitLab