search

akozd / tensorflow_yolo_v3

A well-documented TensorFlow implementation of the YOLOv3 model.
MIT License
44 stars 16 forks source link

Custom training is failing when loading pretrained errors #4

Open paturi1710 opened 5 years ago

paturi1710 commented 5 years ago

Restoring the graph from cocoa pre-trained weights is failing.

Error:

Restoring from checkpoint failed. This is most likely due to a mismatch between the current graph and the graph from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Assign requires shapes of both tensors to match. lhs shape= [21] rhs shape= [255]

SrikarNamburu commented 4 years ago

Were you able to resolve this issue?