fix obvious bugs in training script.

krrish94 / nerf-pytorch

A PyTorch re-implementation of Neural Radiance Fields

Other

887 stars 122 forks source link

fix obvious bugs in training script. #23

Open zhuyinheng opened 3 years ago

zhuyinheng commented 3 years ago

the training script fail to switch model's train/eval mode(model_fine.train() and model_fine.eval()). Since no BN is used in the NeRF, maybe this bug make no difference for now. but may mislead the research based on this implementation.