Module: TensorStream::Trainer

Defined in:

lib/tensor_stream/trainer.rb

Class Method Summary

• .write_graph(graph, path, filename, as_text: true, serializer: TensorStream::Pbtext) ⇒ Object

Class Method Details

.write_graph(graph, path, filename, as_text: true, serializer: TensorStream::Pbtext) ⇒ Object

# File 'lib/tensor_stream/trainer.rb', line 6

def self.write_graph(graph, path, filename, as_text: true, serializer: TensorStream::Pbtext)
  raise "only supports as_text=true for now" unless as_text
  new_filename = File.join(path, filename)
  File.write(new_filename, serializer.new.get_string(graph))
end