e588f6b3985622c42299f7136925aa68d91f1cef,dnc/sparse_memory.py,SparseMemory,reset,#SparseMemory#Any#Any#Any#,84

Before Change

          "read_vectors": cuda(T.zeros(b, r, w).fill_(δ), gpu_id=self.gpu_id),
          "last_used_mem": cuda(T.zeros(b, 1), gpu_id=self.gpu_id).long(),
          "usage": cuda(T.zeros(b, m), gpu_id=self.gpu_id),
          "read_positions": cuda(T.zeros(b, 1, r).fill_(0), gpu_id=self.gpu_id).long()
      }
      hidden = self.rebuild_indexes(hidden, erase=True)
    else:

After Change

        hidden["read_vectors"].data.fill_(δ)
        hidden["last_used_mem"].data.fill_(0)
        hidden["usage"].data.fill_(δ)
        hidden["read_positions"] = cuda(T.arange(0, r).expand(b, 1, r), gpu_id=self.gpu_id).long()
    return hidden

  def write_into_sparse_memory(self, hidden):
    read_vectors = hidden["read_vectors"]

Instances