A PyTorch implementation of NeRF

June 9, 2022 ยท View on GitHub

  • FP16
  • distributed training
CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 train.py --use_fp16 1