1 files changed, 6 insertions, 0 deletions
diff --git a/run.py b/run.py
index f276e86..62e822e 100644
--- a/run.py
+++ b/run.py
@@ -51,7 +51,13 @@ if not os.path.exists('./renders'):
 moduleNetwork = Network(arguments_strModel).cuda()
 tensorOutput = torch.FloatTensor()
 
+index = 0
+
 def process_tree(moduleNetwork, a, b, tensorOutput, steps):
+	global index
+	index += 1
+	if (index % 10) == 0:
+		print("{}...".format(index))
 	tensorInputFirst = torch.FloatTensor(a)
 	tensorInputSecond = torch.FloatTensor(b)
 	process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput)