how to deal with shape mismatch when using init_model?

[hujun100]

Invalid argument: Assign requires shapes of both tensors to match. lhs shape= [250,16384] rhs shape= [150,16384] [[Node: save/Assign_68 = Assign[T=DT_FLOAT, _class=["loc:@G_fc/w"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/gpu:0"]

[Neltherion]

@hujun100 @ZhaoJ9014 Any update on how to fix this ?