Sha256: 3f1b7bc2790c9082d5a9fdb3aa34f0da49ee9434d248b1e7fc4321d1f6177a46
Contents?: true
Size: 422 Bytes
Versions: 8
Compression:
Stored size: 422 Bytes
Contents
require 'tensor_stream/train/gradient_descent_optimizer'
require 'tensor_stream/train/saver'
module TensorStream
module Trainer
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
end
end
Version data entries
8 entries across 8 versions & 1 rubygems