File size: 1,142 Bytes
fe64bad |
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 |
import os
from .base import argparse, add_misc_options, add_cuda_options, adding_cuda
from .tools import save_args
from .dataset import add_dataset_options
from .training import add_training_options
from .checkpoint import construct_checkpointname
def training_parser():
parser = argparse.ArgumentParser()
# misc options
add_misc_options(parser)
# training options
add_training_options(parser)
# dataset options
add_dataset_options(parser)
# model options
add_cuda_options(parser)
opt = parser.parse_args()
# remove None params, and create a dictionnary
parameters = {key: val for key, val in vars(opt).items() if val is not None}
parameters["modelname"] = "recognition"
if "folder" not in parameters:
parameters["folder"] = construct_checkpointname(parameters,
parameters["expname"])
os.makedirs(parameters["folder"], exist_ok=True)
save_args(parameters, folder=parameters["folder"])
adding_cuda(parameters)
return parameters
|