wassname/pytorch-soft-actor-critic
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-soft-actor-critic.git synced 2026-09-09 11:32:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Pranjal Tandon
2019-05-18 22:22:57 +05:30
committed by GitHub
parent ed1d0a55d2
commit 192db2f0a1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+1 -1
View File
@@ -53,7 +53,7 @@ env.seed(args.seed)
# Agent
agent = SAC(env.observation_space.shape[0], env.action_space, args)
writer = SummaryWriter(log_dir='runs/{}_VIREL_{}'.format(datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"), args.env_name))
writer = SummaryWriter(log_dir='runs/{}_SAC_old_{}'.format(datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"), args.env_name))
# Memory
memory = ReplayMemory(args.replay_size)