Contents

#!/usr/bin/env ruby

require "bundler/setup"
require "tensor_stream"
require "tensor_stream/utils/freezer"

if ARGV[0].nil?
  puts "source checkpoint folder not specified"
  puts "usage: model_utils <checkpoint folder> <target yaml>"
  puts "example: model_utils sample_model/ frozen.yml"
  exit(1)
end

if ARGV[1].nil?
  puts "dest YAML file for frozen model not specified"
  puts "usage: model_utils <checkpoint folder> <target yaml>"
  puts "example: model_utils sample_model/ frozen.yml"
  exit(1)
end

sess = TensorStream.session
freezer = TensorStream::Freezer.new
freezer.convert(sess, ARGV[0], ARGV[1])
exit(0)

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
tensor_stream-1.0.9	exe/model_utils
tensor_stream-1.0.8	exe/model_utils
tensor_stream-1.0.7	exe/model_utils
tensor_stream-1.0.6	exe/model_utils
tensor_stream-1.0.5	exe/model_utils
tensor_stream-1.0.4	exe/model_utils
tensor_stream-1.0.3	exe/model_utils
tensor_stream-1.0.2	exe/model_utils
tensor_stream-1.0.1	exe/model_utils