hotfix: remove precrop after the first validation

Author: jeongyw12382

dataloader/sampler.py

@@ -92,6 +92,7 @@ def __iter__(self):
                 np.random.choice(np.arange(image_shape[i, 0] * image_shape[i, 1]), self.batch_size) 
                 for i in range(self.epoch_size - self.precrop_steps)
             ]
+            self.precrop = False
 
         for ((h, w), image_idx, idx) in zip(image_shape, image_choice, idx_choice):
             idx_ret = image_idx * h * w + idx

docs/jaxnerf.md

@@ -8,7 +8,7 @@ In addition, we have implemented the heavy model, dubbed JaxNeRF+, to reproduce
 
 :white_check_mark: DDP support!
 
-:white_check_mark: Large Model (JaxNeRF+)
+:white_check_mark: Large Model (NeRF+)
 
 ## Differences
 This section compares the JaxNeRF and the original NeRF. In addition, we also provide additional details that we have changed in this implementation and their corrsponding reasons.