diff --git a/turbo_alignment/common/logging/clearml.py b/turbo_alignment/common/logging/clearml.py index c429439..1a3e272 100644 --- a/turbo_alignment/common/logging/clearml.py +++ b/turbo_alignment/common/logging/clearml.py @@ -7,7 +7,10 @@ def create_clearml_task(parameters: ClearMLSettings, config: dict[str, Any] | None = None) -> Task: clearml_task = Task.init( - task_name=parameters.task_name, project_name=parameters.project_name, continue_last_task=True # FIXME? + task_name=parameters.task_name, + project_name=parameters.project_name, + continue_last_task=True, + output_uri=False, ) clearml_task.connect_configuration(config, name='HyperParameters') diff --git a/turbo_alignment/common/tf/callbacks/logging.py b/turbo_alignment/common/tf/callbacks/logging.py index c72ebc6..8f90288 100755 --- a/turbo_alignment/common/tf/callbacks/logging.py +++ b/turbo_alignment/common/tf/callbacks/logging.py @@ -70,10 +70,7 @@ def _log(self, logs: dict[str, Any], state: TrainerState) -> None: self._wandb_run.log({**rewritten_logs, 'train/global_step': state.global_step}, step=state.global_step) def _fix_table_type(self, logs: dict[str, Any]) -> dict[str, Any]: - return { - k: wandb.Table(dataframe=v) if isinstance(v, pd.DataFrame) else v - for k, v in logs.items() - } + return {k: wandb.Table(dataframe=v) if isinstance(v, pd.DataFrame) else v for k, v in logs.items()} class ClearMLLoggingCallback(LoggingCallback):