From e9bba85b086abe7ca6db6c81ab7bc1f142872faf Mon Sep 17 00:00:00 2001
From: Yuantao Chen <44546388+Tao-11-chen@users.noreply.github.com>
Date: Wed, 10 May 2023 16:38:34 +0800
Subject: [PATCH] FIx random indices bug (#1900)

* Fix mask bug

* Fix Fix_indices bug
---
 nerfstudio/data/utils/dataloaders.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nerfstudio/data/utils/dataloaders.py b/nerfstudio/data/utils/dataloaders.py
index 7f75b782a..aa30407ea 100644
--- a/nerfstudio/data/utils/dataloaders.py
+++ b/nerfstudio/data/utils/dataloaders.py
@@ -211,6 +211,7 @@ class FixedIndicesEvalDataloader(EvalDataloader):
         self.count = 0
 
     def __iter__(self):
+        self.count = 0
         return self
 
     def __next__(self):