Gwiths

Assign requires shapes of both tensors to match.

2

Hi， @wizyoung I trained the model with my own data, I did not modify the model，but I got the following error： `Invalid argument: Assign requires shapes of both tensors to...