Update README.md

4 years ago · 8b11982846
parent 77d5146ca4
commit 8b11982846
1 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -45,7 +45,7 @@ python ddp_test_nerf.py --config configs/tanks_and_temples/tat_training_truck.tx
                        --render_splits test,camera_path
 ```

-**Note**: due to restriction imposed by torch.gather function, please make sure the number of pixels in each image is divisible by the number of GPUs if you render images parallelly. 
+**Note**: due to restriction imposed by torch.distributed.gather function, please make sure the number of pixels in each image is divisible by the number of GPUs if you render images parallelly. 

 ## Citation
 Plese cite our work if you use the code.