wassname/pytorch-soft-actor-critic
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-soft-actor-critic.git synced 2026-06-27 16:31:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

tensorboard cleanup

Browse Source
This commit is contained in:
pranz24
2020-06-06 09:34:03 +05:30
parent dbbbacc39d
commit e5c349f0b0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+2 -2
View File
@@ -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