File size: 1,522 Bytes
cc46d03 5fef3c0 cc46d03 5fef3c0 cc46d03 5fef3c0 cc46d03 |
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 35 36 37 38 39 40 41 42 43 |
import argparse
def train(path: str, save_to: str, iters: int):
import src.train as trainer
trainer.main(path.strip(), save_to, iters)
# init_train()
def eval(path: str):
import src.eval as evaler
evaler.main(path)
# init_eval()
def chat(path: str, response_path: str):
import src.chat as chatter
chatter.main(path, response_path)
# init_chat
def main():
parser = argparse.ArgumentParser(description="maincmd")
parser.add_argument("--path", type=str, help="path of your model")
subparser = parser.add_subparsers(title="subcmd")
# train
parser_train = subparser.add_parser("train", help="train your model")
parser_train.add_argument("--path", type=str, help="path to your model")
parser_train.add_argument("--save-to", type=str, help="path to save your model")
parser_train.add_argument("--iters", type=int, help="how much iteration your model does for training")
parser_train.set_defaults(func=lambda kwargs: train(kwargs.path, kwargs.save_to, kwargs.iters))
# chat
parser_chat = subparser.add_parser("chat", help="chat with your model")
parser_chat.add_argument("--path", type=str, help="path to your model")
parser_chat.add_argument("--response", type=str, help="path to your responses")
parser_chat.set_defaults(func=lambda kwargs: chat(kwargs.path, kwargs.response))
args = parser.parse_args()
if hasattr(args, 'func'): args.func(args)
elif args.path: eval(args.path)
else: print("invalid argument. exiting program.")
# main():
if __name__ == "__main__": main() |