search

wizyoung / YOLOv3_TensorFlow

Complete YOLO v3 TensorFlow implementation. Support training on your own dataset.
MIT License
1.55k stars 579 forks source link

Assign requires shapes of both tensors to match. #218

Open Gwiths opened 4 years ago

Gwiths commented 4 years ago

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 match. lhs shape= [225] rhs shape= [255] How can i deal with this?

TrinhNC commented 4 years ago

I got similar error when running eval.py, my solution is changing the "--restore_path" to, for example, "./checkpoint/model-epoch_90_step_22750_loss_0.0632_lr_1e-05"

Leeyed commented 3 years ago

I got similar error when running eval.py, my solution is changing the "--restore_path" to, for example, "./checkpoint/model-epoch_90_step_22750_loss_0.0632_lr_1e-05"

add in command and it works :）