# -*- coding: utf-8 -*- import sys import torch from maptor import Paths import maptor.util.argsparser as parser from torch.nn.modules.loss import MSELoss sys.path.append(Paths.LOCALPTRN) from lstmpathmaker.data.IPDataLoader import IPDataLoader args = parser.parse_args() ipdl = IPDataLoader(nagts=args.nagts, tpl=args.map, datasrc=args.datasrc, strt=args.strategy, strt_variant=args.variant, duration=args.duration, soc_name=args.society, inf_exec_id=args.inf_exec_id, sup_exec_id=args.sup_exec_id) ipdl.load_data() viidls = ipdl.domain_data vidls = ipdl.target_data veidls = ipdl.load_veidls() zeros = torch.zeros(veidls.size())
LogProcessor.generate_save_veidls_and_idls(tpl=tpl, strt=strt, variant=variant, nagts=nagts, exec_id=exec_id, duration=duration, log_fp=log_fp, agts_log_fp=agts_log_fp, est_log_fp=est_log_fp, viidls_rep=viidls_rep, vidls_rep=vidls_rep, veidls_rep=veidls_rep, soc_name=soc_name) print("Inputs: ", log_fp, '\n', agts_log_fp, '\n', est_log_fp, '\n') print("Outputs: ", viidls_fp, '\n', veidls_fp, '\n', vidls_fp, '\n') end_time = time.time() print(end_time - start_time, ' s') print('\n') args = argsparser.parse_args() # Executed only if run as a script if __name__ == "__main__": main(strt=args.strategy, nagts=args.nagts, duration=args.duration, exec_id=args.exec_id, tpl=args.map, soc_name=args.society, variant=args.variant)
import sys import time # Adds the current package to the Python's path to run it from anywhere in the # terminal # sys.path.append(os.path.dirname(os.path.abspath(__file__))) from maptor import Paths import maptor.control as control from maptor.util.argsparser import parse_args sys.path.append(Paths.LOCALPYTROL) import pytrol.util.pathformatter as pf args = parse_args() def main(strt: str, variant: str, pop_sizes: list = None, duration: int = 3000, maps: list = None, soc_name: str = None, logs_rep: str = args.dirpath_logs, binpos_rep: str = Paths.LOCALBIN, viidls_rep: str = Paths.LOCALVIIDLS, vidls_rep: str = Paths.LOCALVIDLS): """