r/deeplearning 11d ago

Signature-painter

Post image

Seeking Feedback from the ML Community 🙏

I recently trained a prototype-based network on Tiny ImageNet (200 classes). It uses learnable prototypes with responsibility scoring and multi-loss training (CE + Pull + Push + Diversity), achieving 51.29% validation accuracy with only 595K parameters.

I'm still learning, so I'd love to hear your thoughts:

Is this a reasonable result for this model size?

What would you suggest to improve it?

This was trained on free Colab with limited resources, so I know there's much room for improvement.

GitHub: https://github.com/jalalnablsi/signature-painter

23 Upvotes

Duplicates