Fix restore args in checkpoint.md

[mehdiataei]

I believe restore_args must be equal to {'restore_args': restore_args} instead. Otherwise this will fail with unexpected keyword argument error.