search

microsoft / vs-tools-for-ai

Visual Studio Tools for AI is a free Visual Studio extension to build, test, and deploy deep learning / AI solutions. It seamlessly integrates with Azure Machine Learning for robust experimentation capabilities, including but not limited to submitting data preparation and model training jobs transparently to different compute targets. Additionally, it provides support for custom metrics and run history tracking, enabling data science reproducibility and auditing. Enterprise ready collaboration, allow to securely work on project with other people.
http://aka.ms/vstoolsforai
350 stars 79 forks source link

support for TF frozen models in model inference #14

Open franperezlopez opened 6 years ago

franperezlopez commented 6 years ago

please consider adding support for frozen models when using the model inference tool (given that this support is already present in the model converter)

linmajia commented 6 years ago

@franperezlopez , thank you very much for the feedback.

Currently, we support TensorFlow checkpoint and saved model only. To export to SavedModel format, there is an example at https://github.com/Microsoft/samples-for-ai/blob/master/projects/StyleTransfer/StyleTransferTraining/src/train.py#L218 or you could simply call tf.saved_model.simple_save.