a faster version of config_generator

yanqingmen · Jun 17, 2015 · 3827be2 · 3827be2
1 parent 3dfc7b7
commit 3827be2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/network_maker/config_generator.py b/tools/network_maker/config_generator.py
@@ -246,7 +246,7 @@ def ConvFactory(nchannel, kernel_size=1, pad=0, stride = 1, bn = True, act = "re
 
 def DFS(layer, table, seq):
  out_layers = graph[layer]
- for l in out_layers:
+ for l in out_layers - table:
  DFS(l, table, seq)
  if layer not in table:
  table.add(layer)