diff --git a/src/testbed_nerf.cu b/src/testbed_nerf.cu index cc054f828d981e1efae0f85f3ec1bc07bbe14f7f..0f9f31e3d2f72b00463a6556228651c9ef76cbcf 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