diff --git a/ernerf/nerf_triplane/utils.py b/ernerf/nerf_triplane/utils.py index fa0b562..ac75ca1 100644 --- a/ernerf/nerf_triplane/utils.py +++ b/ernerf/nerf_triplane/utils.py @@ -550,7 +550,7 @@ def update(self, preds, truths): lms_pred = lms_pred[48:68] lms_truth = lms_truth[48:68] - # avarage + # average lms_pred = lms_pred - lms_pred.mean(0) lms_truth = lms_truth - lms_truth.mean(0) @@ -1513,4 +1513,4 @@ def load_checkpoint(self, checkpoint=None, model_only=False): self.scaler.load_state_dict(checkpoint_dict['scaler']) self.log("[INFO] loaded scaler.") except: - self.log("[WARN] Failed to load scaler.") \ No newline at end of file + self.log("[WARN] Failed to load scaler.")