You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

ModelSaver.cs 741 B

1234567891011121314151617181920212223242526
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Text;
  4. using Tensorflow.Keras.Engine;
  5. using Tensorflow.Train;
  6. using Tensorflow.Training.Saving.SavedModel;
  7. namespace Tensorflow.ModelSaving
  8. {
  9. public class ModelSaver
  10. {
  11. public void save(Trackable obj, string export_dir, SaveOptions options = null)
  12. {
  13. var saved_model = new SavedModel();
  14. var meta_graph_def = new MetaGraphDef();
  15. saved_model.MetaGraphs.Add(meta_graph_def);
  16. _build_meta_graph(obj, export_dir, options, meta_graph_def);
  17. }
  18. void _build_meta_graph(Trackable obj, string export_dir, SaveOptions options,
  19. MetaGraphDef meta_graph_def = null)
  20. {
  21. }
  22. }
  23. }