Update model_non_linear_3DMM.py

booler · Jul 30, 2019 · 1c0c7d3 · 1c0c7d3
1 parent 8154245
commit 1c0c7d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model_non_linear_3DMM.py b/model_non_linear_3DMM.py
@@ -401,8 +401,8 @@ def train(self, config):
          # Training data
         self.setupTrainingData()
 
-        valid_idx = range(self.images_300W.shape[0])
-        print("Valid images %d/%d" % ( len(valid_300W_idx), self.images_300W.shape[0] ))
+        valid_idx = range(self.image_filenames.shape[0])
+        print("Valid images %d/%d" % ( len(valid_idx), self.image_filenames.shape[0] ))