HaoYizhi/IIabm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

first normal run

Browse Source
This commit is contained in:
HaoYizhi
2023-03-16 21:38:57 +08:00
parent 1072d4e2b0
commit 6489ecff90
9 changed files with 50 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@@ -23,7 +23,8 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser(description='setting')
parser.add_argument('--exp', type=str, default='test')
parser.add_argument('--job', type=int, default='3')
parser.add_argument('--reset', type=int, default='0')
parser.add_argument('--reset_sample', type=int, default='0')
parser.add_argument('--reset_db', type=bool, default=False)
args = parser.parse_args()
assert args.job >= 1, 'Number of jobs should >= 1'
@@ -35,11 +36,11 @@ if __name__ == '__main__':
yaml.dump({'db_name_prefix': args.exp}, file)
from controller_db import ControllerDB
controller_db = ControllerDB(args.exp, reset_flag=args.reset)
controller_db = ControllerDB(args.exp, reset_flag=args.reset_sample)
# controller_db.reset_db()
# force drop
controller_db.reset_db(force_drop=True)
controller_db.reset_db(force_drop=args.reset_db)
controller_db.prepare_list_sample()