hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Apply datetime_tag() in high-level examples

Browse Source
This commit is contained in:
Dominik Jain 2023-10-26 11:24:55 +02:00
parent d684dae6cd
commit c613557740
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/atari/atari_ppo_hl.py
View File

@ -1,6 +1,5 @@
#!/usr/bin/env python3
import datetime
import os
from collections.abc import Sequence
@ -20,6 +19,7 @@ from tianshou.highlevel.params.policy_wrapper import (
PolicyWrapperFactoryIntrinsicCuriosity,
)
from tianshou.utils import logging
from tianshou.utils.logging import datetime_tag
def main(
@ -54,8 +54,7 @@ def main(
icm_reward_scale: float = 0.01,
icm_forward_loss_weight: float = 0.2,
):
now = datetime.datetime.now().strftime("%y%m%d-%H%M%S")
log_name = os.path.join(task, "ppo", str(experiment_config.seed), now)
log_name = os.path.join(task, "ppo", str(experiment_config.seed), datetime_tag())
sampling_config = SamplingConfig(
num_epochs=epoch,

5
examples/mujoco/mujoco_sac_hl.py
View File

@ -1,6 +1,5 @@
#!/usr/bin/env python3
import datetime
import os
from collections.abc import Sequence
@ -13,6 +12,7 @@ from tianshou.highlevel.experiment import (
from tianshou.highlevel.params.alpha import AutoAlphaFactoryDefault
from tianshou.highlevel.params.policy_params import SACParams
from tianshou.utils import logging
from tianshou.utils.logging import datetime_tag
def main(
@ -37,8 +37,7 @@ def main(
training_num: int = 1,
test_num: int = 10,
):
now = datetime.datetime.now().strftime("%y%m%d-%H%M%S")
log_name = os.path.join(task, "sac", str(experiment_config.seed), now)
log_name = os.path.join(task, "sac", str(experiment_config.seed), datetime_tag())
sampling_config = SamplingConfig(
num_epochs=epoch,