Fix incorrectly sized gradient buffer (reduces memory footprint)

NVlabs · Mar 1, 2022 · 36278ee · 36278ee
1 parent 8ce250e
commit 36278ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/testbed_nerf.cu b/src/testbed_nerf.cu
@@ -2663,7 +2663,7 @@ void Testbed::train_nerf_step(uint32_t target_batch_size, uint32_t n_rays_per_ba
 		max_samples * floats_per_coord,
 		max_samples,
 		std::max(target_batch_size, max_samples) * padded_output_width,
-		std::max(target_batch_size, max_samples) * 4,
+		target_batch_size * padded_output_width,
 		target_batch_size * padded_output_width,
 		target_batch_size * floats_per_coord,
 		target_batch_size