mirror of
https://github.com/unanmed/ginka-generator.git
synced 2026-05-14 04:41:12 +08:00
13 lines
625 B
Python
13 lines
625 B
Python
import argparse
|
|
|
|
def parse_arguments(from_default: str, train_default: str, val_default: str):
|
|
parser = argparse.ArgumentParser(description="training codes")
|
|
parser.add_argument("--resume", type=bool, default=False)
|
|
parser.add_argument("--from_state", type=str, default=from_default)
|
|
parser.add_argument("--load_optim", type=bool, default=False)
|
|
parser.add_argument("--train", type=str, default=train_default)
|
|
parser.add_argument("--validate", type=str, default=val_default)
|
|
parser.add_argument("--epochs", type=int, default=150)
|
|
args = parser.parse_args()
|
|
return args
|