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

2 entries across 2 versions & 1 rubygems

Version	Path
tensor_stream-1.0.0	exe/model_utils
tensor_stream-1.0.0.pre.rc1	exe/model_utils