8dc0a7cd41a23aa58ee44bbadd14b8196a886c96,tasks/copy_task.py,,,#,93

Before Change

    input_data, target_output = generate_data(batch_size, random_length, args.input_size, args.cuda)
    // input_data = input_data.transpose(0, 1).contiguous()
    target_output = target_output.transpose(0, 1).contiguous()

    output, (chx, mhx, rv), v = rnn(input_data, None, pass_through_memory=True)
    // dncs operate batch first
    output = output.transpose(0, 1)

After Change

    if rnn.debug:
      output, (chx, mhx, rv), v = rnn(input_data, None, pass_through_memory=True)
    else:
      output, (chx, mhx, rv) = rnn(input_data, None, pass_through_memory=True)

    loss = criterion((output), target_output)

    loss.backward()

Instances