Check failed: (reporter->AssertEQ(data->shape[data->shape.size() - 1], weight->shape[1])) is false: DenseRel: input dimension doesn't match, data shape=[1, 512], weight shape=[512, 1000]

I was not able to reproduce the error. I am using torch 1.9.0.post2. Do you still have an issue?