ResNet50 Keras: Export to saved_model as stated by the documentation

HabanaAI / Model-References

Reference models for Intel(R) Gaudi(R) AI Accelerator

155 stars 81 forks source link

ResNet50 Keras: Export to saved_model as stated by the documentation #8

Open levzlotnik opened 3 years ago

levzlotnik commented 3 years ago

The flag enable_checkpoint_and_export states to export to saved_model format after training, although running the script with this flag didn't actually export the saved_model.
Added the fix to correct behavior in resnet_ctl_imagenet_main.py and a flag to the horovod launcher script.