537e5b549a9c4eeb3fe2776028b553ef36fadd88,rnns/fastrnns/test.py,,test_rnns,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,27

Before Change

    control_output = control_outputs[0]
    experim_output = experim_outputs[0]
    grad = torch.randn_like(control_output)
    control_grads = torch.autograd.grad([control_output], control_params, grad)
    experim_grads = torch.autograd.grad([experim_output], experim_params, grad)
    assertEqual(experim_grads, control_grads)

After Change

    print("Checking outputs...")
    control_outputs = control.forward(*control.inputs)
    experim_outputs = experim.forward(*experim.inputs)
    assertEqual(experim_outputs, control_outputs)

    print("Checking grads...")
    assert control.backward_setup is not None
    assert experim.backward_setup is not None
    assert control.backward is not None
    assert experim.backward is not None
    control_backward_inputs = control.backward_setup(control_outputs, seed)
    experim_backward_inputs = experim.backward_setup(experim_outputs, seed)

    control.backward(*control_backward_inputs)
    experim.backward(*experim_backward_inputs)

Instances