diff --git a/main.py b/main.py index e6ac07b..85973de 100644 --- a/main.py +++ b/main.py @@ -56,8 +56,8 @@ env.seed(args.seed) # Agent agent = SAC(env.observation_space.shape[0], env.action_space, args) -#TesnorboardX -writer = SummaryWriter(logdir='runs/{}_SAC_{}_{}_{}'.format(datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"), args.env_name, +#Tesnorboard +writer = SummaryWriter('runs/{}_SAC_{}_{}_{}'.format(datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"), args.env_name, args.policy, "autotune" if args.automatic_entropy_tuning else "")) # Memory