search

leondgarse / Keras_insightface

Insightface Keras implementation
MIT License
234 stars 56 forks source link

Keras Tuner Support #129

Open saioru opened 8 months ago

saioru commented 8 months ago

Quite new to Keras trainings, would like to ask if its applicable to leverage keras-tuner on optimizing the hyperparameters in finetune and/or training of the models as this library is a Keras implementation of deepinsight/insightface.

leondgarse commented 8 months ago

Ya, it's possible applying keras-tuner here, but needs some works. Like modifying train.py for training hyper parameters, and model architectures in backbones, adding hp configurations. But never have I tried that, not sure the affect on results... You may also refer All from scratch #71 for a basic usage with keras and tensorflow alone.

saioru commented 8 months ago

Ya, it's possible applying keras-tuner here, but needs some works. Like modifying train.py for training hyper parameters, and model architectures in backbones, adding hp configurations. But never have I tried that, not sure the affect on results... You may also refer All from scratch #71 for a basic usage with keras and tensorflow alone.

Cheers for the heads up on this, will be conducting a few experiments and modifications to train.py and hp configs. Any suggested amount of data prep required for fine-tuning ResNet34 with Arcface?

leondgarse commented 8 months ago

You may start with 100 classes and 50 images per class. Just a reasonable start point. :)