def main():
import six.moves._thread as thread
import threading
thread.stack_size(128 * 1024 * 1024)
sys.setrecursionlimit(0x3fffffff)
main_thread = threading.Thread(target=cli_main)
main_thread.start()
main_thread.join()
def main():
global return_value
import six.moves._thread as thread
import threading
thread.stack_size(128 * 1024 * 1024)
sys.setrecursionlimit(0x3fffffff)
main_thread = threading.Thread(target=cli_main)
main_thread.start()
main_thread.join()
if not return_value:
sys.exit(-1)
# Create TAR for uploader
from nnabla.utils.cli.uploader import create_tar_command
subparser = subparsers.add_parser('create_tar')
subparser.add_argument('source', help='CSV dataset')
subparser.add_argument('destination', help='TAR filename')
subparser.set_defaults(func=create_tar_command)
# Extract nnp file
from nnabla.utils.cli.extract import extract_command
subparser = subparsers.add_parser('extract')
subparser.add_argument(
'-l', '--list', help='list contents.', action='store_true')
subparser.add_argument(
'-x', '--extract', help='extract contents to current dir.', action='store_true')
subparser.add_argument('nnp', help='nnp filename')
subparser.set_defaults(func=extract_command)
args = parser.parse_args()
args.func(args)
if __name__ == '__main__':
import six.moves._thread as thread
import threading
thread.stack_size(128 * 1024 * 1024)
sys.setrecursionlimit(0x3fffffff)
main_thread = threading.Thread(target=main)
main_thread.start()
main_thread.join()
def main():
parser = argparse.ArgumentParser(
description='Command line interface ' +
'for NNabla({})'.format(_nnabla_version()))
parser.add_argument('-m',
'--mpi',
help='exec with mpi.',
action='store_true')
subparsers = parser.add_subparsers()
from nnabla.utils.cli.train import add_train_command
add_train_command(subparsers)
from nnabla.utils.cli.forward import add_infer_command, add_forward_command
add_infer_command(subparsers)
add_forward_command(subparsers)
from nnabla.utils.cli.encode_decode_param import add_decode_param_command, add_encode_param_command
add_encode_param_command(subparsers)
add_decode_param_command(subparsers)
from nnabla.utils.cli.profile import add_profile_command
add_profile_command(subparsers)
from nnabla.utils.cli.conv_dataset import add_conv_dataset_command
add_conv_dataset_command(subparsers)
from nnabla.utils.cli.compare_with_cpu import add_compare_with_cpu_command
add_compare_with_cpu_command(subparsers)
from nnabla.utils.cli.create_image_classification_dataset import add_create_image_classification_dataset_command
add_create_image_classification_dataset_command(subparsers)
from nnabla.utils.cli.create_object_detection_dataset import add_create_object_detection_dataset_command
add_create_object_detection_dataset_command(subparsers)
from nnabla.utils.cli.uploader import add_upload_command
add_upload_command(subparsers)
from nnabla.utils.cli.uploader import add_create_tar_command
add_create_tar_command(subparsers)
from nnabla.utils.cli.convert import add_convert_command
add_convert_command(subparsers)
from nnabla.utils.cli.func_info import add_function_info_command
add_function_info_command(subparsers)
from nnabla.utils.cli.optimize_pb_model import add_optimize_pb_model_command
add_optimize_pb_model_command(subparsers)
from nnabla.utils.cli.plot import (add_plot_series_command,
add_plot_timer_command)
add_plot_series_command(subparsers)
add_plot_timer_command(subparsers)
from nnabla.utils.cli.draw_graph import add_draw_graph_command
add_draw_graph_command(subparsers)
# Version
subparser = subparsers.add_parser('version',
help='Print version and build number.')
subparser.set_defaults(func=version_command)
print('NNabla command line interface ({})'.format(_nnabla_version()))
args = parser.parse_args()
import nnabla.utils.callback as callback
r = callback.alternative_cli(args)
if r is not None:
return r
global return_value
import six.moves._thread as thread
import threading
thread.stack_size(8 * 1024 * 1024)
sys.setrecursionlimit(1024 * 1024)
main_thread = threading.Thread(target=cli_main, args=(parser, args))
main_thread.start()
main_thread.join()
if not return_value:
sys.exit(-1)
