Apply datetime_tag() in high-level examples
This commit is contained in:
parent
d684dae6cd
commit
c613557740
@ -1,6 +1,5 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import datetime
|
|
||||||
import os
|
import os
|
||||||
from collections.abc import Sequence
|
from collections.abc import Sequence
|
||||||
|
|
||||||
@ -20,6 +19,7 @@ from tianshou.highlevel.params.policy_wrapper import (
|
|||||||
PolicyWrapperFactoryIntrinsicCuriosity,
|
PolicyWrapperFactoryIntrinsicCuriosity,
|
||||||
)
|
)
|
||||||
from tianshou.utils import logging
|
from tianshou.utils import logging
|
||||||
|
from tianshou.utils.logging import datetime_tag
|
||||||
|
|
||||||
|
|
||||||
def main(
|
def main(
|
||||||
@ -54,8 +54,7 @@ def main(
|
|||||||
icm_reward_scale: float = 0.01,
|
icm_reward_scale: float = 0.01,
|
||||||
icm_forward_loss_weight: float = 0.2,
|
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), datetime_tag())
|
||||||
log_name = os.path.join(task, "ppo", str(experiment_config.seed), now)
|
|
||||||
|
|
||||||
sampling_config = SamplingConfig(
|
sampling_config = SamplingConfig(
|
||||||
num_epochs=epoch,
|
num_epochs=epoch,
|
||||||
|
@ -1,6 +1,5 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import datetime
|
|
||||||
import os
|
import os
|
||||||
from collections.abc import Sequence
|
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.alpha import AutoAlphaFactoryDefault
|
||||||
from tianshou.highlevel.params.policy_params import SACParams
|
from tianshou.highlevel.params.policy_params import SACParams
|
||||||
from tianshou.utils import logging
|
from tianshou.utils import logging
|
||||||
|
from tianshou.utils.logging import datetime_tag
|
||||||
|
|
||||||
|
|
||||||
def main(
|
def main(
|
||||||
@ -37,8 +37,7 @@ def main(
|
|||||||
training_num: int = 1,
|
training_num: int = 1,
|
||||||
test_num: int = 10,
|
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), datetime_tag())
|
||||||
log_name = os.path.join(task, "sac", str(experiment_config.seed), now)
|
|
||||||
|
|
||||||
sampling_config = SamplingConfig(
|
sampling_config = SamplingConfig(
|
||||||
num_epochs=epoch,
|
num_epochs=epoch,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user