File size: 859 Bytes
4c62147 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import sys
import argparse
from torchtask.utils import cmd
from torchtask.nn import optimizer, lrer
from torchtask.nn.func import pytorch_support
def create_parser():
parser = argparse.ArgumentParser(description='TorchTask Script Parser')
optimizer.add_parser_arguments(parser)
lrer.add_parser_arguments(parser)
return parser
def run_script(config, proxy_file, proxy_class):
# TorchTask requires PyTorch >= 1.0.0
pytorch_support(required_version='1.0.0', info_str='TorchTask')
# help information
if len(sys.argv) > 1 and sys.argv[1] in ['help', '--help', 'h', '-h']:
config['h'] = True
# create parser and parse args from config
parser = create_parser()
proxy_file.add_parser_arguments(parser)
args = cmd.parse_args(parser, config)
task_proxy = proxy_class(args)
task_proxy.run()
|