| 22 | return model_g, dim |
| 23 | |
| 24 | def log_set(kwargs): |
| 25 | source_data = kwargs["source_data"] |
| 26 | target_data = kwargs["target_data"] |
| 27 | network = kwargs["network"] |
| 28 | conf_file = kwargs["config_file"] |
| 29 | script_name = kwargs["script_name"] |
| 30 | multi = kwargs["multi"] |
| 31 | #args = kwargs["args"] |
| 32 | |
| 33 | target_data = os.path.splitext(os.path.basename(target_data))[0] |
| 34 | logname = "{file}_{source}2{target}_{network}_hp_{hp}".format(file=script_name.replace(".py", ""), |
| 35 | source=source_data.split("_")[1], |
| 36 | target=target_data, |
| 37 | network=network, |
| 38 | hp=str(multi)) |
| 39 | logname = os.path.join("record", kwargs["exp_name"], |
| 40 | os.path.basename(conf_file).replace(".yaml", ""), logname) |
| 41 | if not os.path.exists(os.path.dirname(logname)): |
| 42 | os.makedirs(os.path.dirname(logname)) |
| 43 | print("record in %s " % logname) |
| 44 | import logging |
| 45 | logger = logging.getLogger(__name__) |
| 46 | logging.basicConfig(filename=logname, format="%(message)s") |
| 47 | logger.setLevel(logging.INFO) |
| 48 | logger.info("{}_2_{}".format(source_data, target_data)) |
| 49 | return logname |
| 50 | |
| 51 | |
| 52 | def save_model(model_g, model_c1, model_c2, save_path): |