Browse Source

move isodate to expname

refactoring
sp 10 months ago
parent
commit
d2aa224bc8
  1. 5
      examples/shields/rl/utils.py

5
examples/shields/rl/utils.py

@ -134,9 +134,10 @@ class MiniGridShieldHandler(ShieldHandler):
return self.__create_shield_dict() return self.__create_shield_dict()
def expname(args): def expname(args):
return f"{args.env}_{args.shielding}_{args.shield_comparison}_{args.shield_value}"
return f"{datetime.datetime.now().strftime('%Y%m%dT%H%M%S')}_{args.env}_{args.shielding}_{args.shield_comparison}_{args.shield_value}"
def create_log_dir(args): def create_log_dir(args):
log_dir = f"{args.log_dir}/{datetime.datetime.now().strftime('%Y%m%dT%H%M%S')}_{expname(args)}"
log_dir = f"{args.log_dir}/{expname(args)}"
os.makedirs(log_dir, exist_ok=True) os.makedirs(log_dir, exist_ok=True)
return log_dir return log_dir

Loading…
Cancel
Save