insightface
insightface copied to clipboard
Feature: model save interval = config.verbose and code sync script
I just slightly modified callbacks.py and train_memory.py so that the number of steps between saving the model is the same as config.verbose parameter; so the model will be saved after each validation. It's worth also checking that the validation accuracy improved since the last time the model was saved.
I included a simple script for syncing the code across all the compute nodes, it uses the host/host_x files to automatically rsync the code.
Thank you.