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

Add option to disable file logging

Browse Source
This commit is contained in:
Dominik Jain 2023-10-27 18:59:43 +02:00
parent a3dbe90515
commit 5952993cfe
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tianshou/highlevel/experiment.py
View File

@ -108,6 +108,9 @@ class ExperimentConfig:
in this directory based on the run's experiment name""" in this directory based on the run's experiment name"""
persistence_enabled: bool = True persistence_enabled: bool = True
"""Whether persistence is enabled, allowing files to be stored""" """Whether persistence is enabled, allowing files to be stored"""
log_file_enabled: bool = True
"""Whether to write to a log file; has no effect if `persistence_enabled` is False.
Disable this if you have externally configured log file generation."""
policy_persistence_mode: PolicyPersistence.Mode = PolicyPersistence.Mode.POLICY policy_persistence_mode: PolicyPersistence.Mode = PolicyPersistence.Mode.POLICY
"""Controls the way in which the policy is persisted""" """Controls the way in which the policy is persisted"""
@ -205,7 +208,7 @@ class Experiment(ToStringMixin):
with logging.FileLoggerContext( with logging.FileLoggerContext(
os.path.join(persistence_dir, self.LOG_FILENAME), os.path.join(persistence_dir, self.LOG_FILENAME),
enabled=use_persistence, enabled=use_persistence and self.config.log_file_enabled,
): ):
# log initial information # log initial information
log.info(f"Running experiment (name='{experiment_name}'):\n{self.pprints()}") log.info(f"Running experiment (name='{experiment_name}'):\n{self.pprints()}")