Sha256: 205bb665730f04440ff3264cd58d12146c9e948b2f012b003d044cace23742eb
Contents?: true
Size: 904 Bytes
Versions: 1
Compression:
Stored size: 904 Bytes
Contents
# frozen_string_literal: true require 'onnx-chainer' require 'json' require 'optparse' require 'pathname' module OnnxChainer class CLI def self.start(argv) new(argv).run end 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 @onnx_path = argv.pop @parser.parse!(argv) end 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 end end
Version data entries
1 entries across 1 versions & 1 rubygems
Version | Path |
---|---|
onnx-red-chainer-0.1.0 | lib/onnx-chainer/cli.rb |