dhlab-epfl/dhSegment

Model is not exported after training

stefan-it opened this issue · 2 comments

Hi :)

I trained a model from scratch for ~ 10 epochs. Unfortunately, the model is not exported (the export folder is empty).

Is there any workaround to create an exported model from the checkpoints?

Thanks in advance :)

Hi,

There should be a model exported at the end of training, no matter how many epochs... Have you tried training for more than 10 epochs ? You can continue training you model using python train.py with config.json restore_model=True.
You can also try to change in config.json the parameter evaluate_every_epoch to a lower value, so that your model gets exported after the evaluation.

Otherwise you can use tf.Estimator to export your model (see the doc here).

Closing because of inactivity.