Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
DL4J: Improve exception for invalid YOLO config #6584
It seems to be a common error where users don't correctly size the last convolutional layer in their network before the YOLO layer. Currently, they get this obscure shape error. Better: an exception that tells them exactly what the likely cause is, and what they should do to fix it.