Found another typo in my training captions, so here's a fixed version.
Trained on pictures of burr and few other puzzles, but should be usable for other things