class OnnxChainer::CLI
Public Class Methods
new(argv)
click to toggle source
# File lib/onnx-chainer/cli.rb, line 14 def initialize(argv) @argv = argv.dup @parser = OptionParser.new do |opts| opts.banner = 'onnx-red-chainer [OPTIONS] FILE' opts.version = VERSION opts.on('-o', '--output_dir=OUTPUT_DIR', 'output path') { |v| @output = v } opts.on('-m', '--model_name=MODEL_NAME', 'Model name') { |v| @model_name = v } opts.on('-h', '--help', 'show help') { @help = true } end @parser.parse!(argv) @onnx_path = argv.pop end
start(argv)
click to toggle source
# File lib/onnx-chainer/cli.rb, line 10 def self.start(argv) new(argv).run end
Public Instance Methods
run()
click to toggle source
# File lib/onnx-chainer/cli.rb, line 28 def run if @help || @argv.empty? puts @parser.help else graph = OnnxChainer.parse_file(@onnx_path) graph.export(output_dir: @output, model_name: @model_name) end end