This commit is contained in:
wls2002
2023-05-06 18:33:30 +08:00
parent 73ac1bcfe0
commit 14fed83193
10 changed files with 206 additions and 22 deletions

View File

@@ -27,7 +27,7 @@ def create_forward_function(nodes: NDArray, connections: NDArray,
"""
if debug:
cal_seqs = topological_sort(nodes, connections)
cal_seqs = topological_sort_debug(nodes, connections)
return lambda inputs: forward_single_debug(inputs, N, input_idx, output_idx,
cal_seqs, nodes, connections)