7d9d7698e50296de0b9a09e5bb3d7bfde79ba789,examples/equivariance_error/main.py,,,#,38

Before Change

    return torch.cat([x[:, :, :, n:], x[:, :, :, :n]], dim=3)

// Create random input
x = torch.autograd.Variable(torch.randn(1, 12, 128, 128), volatile=True).cuda() // [batch, feature, beta, alpha]

y = phi(x)
y1 = rot(phi(x), angle=45)
y2 = phi(rot(x, angle=45))

relative_error = torch.std(y1.data - y2.data) / torch.std(y.data)

After Change

    return torch.cat([x[:, :, :, n:], x[:, :, :, :n]], dim=3)

// Create random input
x = torch.randn(1, 12, 128, 128).cuda() // [batch, feature, beta, alpha]

y = phi(x)
y1 = rot(phi(x), angle=45)
y2 = phi(rot(x, angle=45))

relative_error = torch.std(y1.data - y2.data) / torch.std(y.data)

Instances