Module: TensorStream::Trainer

Extended by:

TensorStream::Train::Utils

Defined in:

lib/tensor_stream/trainer.rb

Class Method Summary

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

Methods included from TensorStream::Train::Utils

create_global_step, get_global_step

Class Method Details

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

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

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