diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:41:10 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-25 15:41:10 +0200 |
| commit | 26a08621599ed6a30137c41b32a4b8efac5ff97e (patch) | |
| tree | bfb571dd49c081ec7525fd9fdcb5418e2a90220e /run.py | |
| parent | 28cdb53b65c11254a1855657af73a623d2cc290a (diff) | |
go hey
Diffstat (limited to 'run.py')
| -rw-r--r-- | run.py | 6 |
1 files changed, 6 insertions, 0 deletions
@@ -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) |