Problem with convert.py

[Hendry2182]

Hi Jeremy, I am running convert.py and get this error:
InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [7,1,1,16] rhs shape= [7,1,513,16]
[[Node: save/Assign_1 = Assign[T=DT_FLOAT, _class=["loc:@Encoder/Conv2d-0/Conv2d-0/kernel"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/cpu:0"](Encoder/Conv2d-0/Conv2d-0/kernel, save/RestoreV2_1)]]

It is because I use CPU, so I change all the NCHW to NHWC to support CPU version.