iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add print into baselines

Browse Source
This commit is contained in:
D-X-Y
2021-03-30 12:44:54 +00:00
parent 28a6c57f6e
commit e1224d6883
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/trading/baselines.py
View File

@@ -212,6 +212,7 @@ if __name__ == "__main__":
) )
elif len(args.alg) > 1: elif len(args.alg) > 1:
assert args.shared_dataset, "Must allow share dataset" assert args.shared_dataset, "Must allow share dataset"
pprint(args)
configs = [ configs = [
update_gpu(update_market(alg2configs[name], args.market), args.gpu) update_gpu(update_market(alg2configs[name], args.market), args.gpu)
for name in args.alg for name in args.alg
@@ -222,6 +223,7 @@ if __name__ == "__main__":
pprint(dataset_config) pprint(dataset_config)
pprint(dataset) pprint(dataset)
for alg_name, config in zip(args.alg, configs): for alg_name, config in zip(args.alg, configs):
print("Run {:} over {:}".format(alg_name, args.alg))
for irun in range(args.times): for irun in range(args.times):
run_exp( run_exp(
config.get("task"), config.get("task"),