diff options
Diffstat (limited to 'training/run_experiment.py')
-rw-r--r-- | training/run_experiment.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/training/run_experiment.py b/training/run_experiment.py index f46803f..1e79461 100644 --- a/training/run_experiment.py +++ b/training/run_experiment.py @@ -15,7 +15,7 @@ import wandb SEED = 4711 CONFIGS_DIRNAME = Path(__file__).parent.resolve() / "configs" -LOGS_DIRNAME = Path(__file__).parent.resolve() / "runs" / "logs" +LOGS_DIRNAME = Path(__file__).parent.resolve() / "logs" def _configure_logging(log_dir: Optional[Path], verbose: int = 0) -> None: @@ -67,7 +67,7 @@ def _configure_callbacks( def _configure_logger( - network: Type[nn.Module], args: Dict, log_dir: str, use_wandb: bool + network: Type[nn.Module], args: Dict, log_dir: str, use_wandb: bool ) -> Type[pl.loggers.LightningLoggerBase]: """Configures lightning logger.""" if use_wandb: @@ -124,7 +124,6 @@ def run( _configure_logging(None, verbose=verbose) logger.info("Starting experiment...") - # Seed everything in the experiment. logger.info(f"Seeding everthing with seed={SEED}") pl.utilities.seed.seed_everything(SEED) |