def main(): """main function""" if len(sys.argv) > 1: util.USER_ID = sys.argv[1] def mk_for_userid(): util.CACHE_DIR = os.path.join(util.CACHE_DIR, util.USER_ID) + '/' util.RES_DIR = os.path.join(util.RES_DIR, util.USER_ID) + '/' util.LOG_DIR = os.path.join(util.LOG_DIR, util.USER_ID) + '/' util.MODEL_DIR = os.path.join(util.MODEL_DIR, util.USER_ID) + '/' util.SUMMARIES_DIR = os.path.join(util.SUMMARIES_DIR, util.USER_ID) + '/' # util.TRAIN_YAML = util.USER_ID + ".yaml" mk_for_userid() if len(sys.argv) > 2: util.INFER = sys.argv[2] # flag = True util.check_tensorflow_version() util.check_and_mkdir() config = load_yaml() check_config(config) hparams = create_hparams(config) hparams = create_hparams(config) log = Log(hparams) hparams.logger = log.logger print(hparams.values()) import train train.train(hparams)
def main(): """main function""" # flag = True util.check_tensorflow_version() util.check_and_mkdir() config = load_yaml() check_config(config) hparams = create_hparams(config) log = Log(hparams) hparams.logger = log.logger print(hparams.values()) train.train(hparams)
def main(): """main function""" # flag = True util.check_tensorflow_version() util.check_and_mkdir() #util.TRAIN_YAML = yaml config = load_yaml() check_config(config) hparams = create_hparams(config) log = Log(hparams) hparams.logger = log.logger train.train(hparams)
def main(): """main function""" # flag = True util.check_tensorflow_version() util.check_and_mkdir() #util.TRAIN_YAML = yaml config = load_yaml() check_config(config) hparams = create_hparams(config) print(hparams.values()) log = Log(hparams) hparams.logger = log.logger train.train(hparams)
def main(): """main function""" # flag = True import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1" util.check_tensorflow_version() util.check_and_mkdir() #util.TRAIN_YAML = yaml config = load_yaml() check_config(config) hparams = create_hparams(config) #(name, value) print(hparams.values()) log = Log(hparams) hparams.logger = log.logger train.train(hparams)