4079060908ec43d01e86665792b165269c9545b4,dnc/dnc.py,DNC,_layer_forward,#DNC#Any#Any#Any#,167

Before Change

      else:
        read_vecs, mhx = self.memories[layer](ξ, mhx)
        if self.debug:
          mem_debug.append(mhx["memory"][0].data.cpu().numpy())
      read_vectors[time] = read_vecs.view(-1, self.w * self.r)

      // get the final output for this time step

After Change

    read_vectors = [0] * max_length

    if self.debug:
      mem_debug = {
        "memory": [],
        "link_matrix": [],
        "precedence": [],
        "read_weights": [],
        "write_weights": [],
        "usage_vector": [],
      }

    for time in range(max_length):
      // pass through controller
      layer_input = input[time]

Instances