From 26a08621599ed6a30137c41b32a4b8efac5ff97e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 25 Jun 2018 15:41:10 +0200 Subject: go hey --- run.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'run.py') 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) -- cgit v1.2.3-70-g09d2