When I was debugging the code, I found a problem: by modifying target_sample_batch_size, init_batch_size, render_step_size does not change the GPU memory footprint, only by modifying the network size, such as lowering net_depth, net_width will the GPU memory footprint be reduced.
Why does lowering the batch size not lead to a reduction in GPU memory usage?
When I was debugging the code, I found a problem: by modifying
target_sample_batch_size,init_batch_size,render_step_sizedoes not change the GPU memory footprint, only by modifying the network size, such as loweringnet_depth,net_widthwill the GPU memory footprint be reduced. Why does lowering the batch size not lead to a reduction in GPU memory usage?