Contents

require 'tensor_stream/train/slot_creator'
require 'tensor_stream/train/optimizer'
require 'tensor_stream/train/gradient_descent_optimizer'
require 'tensor_stream/train/momentum_optimizer'
require 'tensor_stream/train/adam_optimizer'
require 'tensor_stream/train/adadelta_optimizer'
require 'tensor_stream/train/adagrad_optimizer'
require 'tensor_stream/train/rmsprop_optimizer'
require 'tensor_stream/train/saver'
require 'tensor_stream/train/learning_rate_decay'

module TensorStream
  module Trainer
    extend TensorStream::Train::Utils
    extend TensorStream::Train::LearningRateDecay

    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

4 entries across 4 versions & 1 rubygems

Version	Path
tensor_stream-0.9.8	lib/tensor_stream/trainer.rb
tensor_stream-0.9.7	lib/tensor_stream/trainer.rb
tensor_stream-0.9.6	lib/tensor_stream/trainer.rb
tensor_stream-0.9.5	lib/tensor_stream/trainer.rb