mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix new config in visualizer (#1390)

Browse Source
This commit is contained in:
Mashiro 2023-10-26 15:31:03 +08:00 committed by GitHub
parent c65187c6b8
commit e43bbb5e03
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mmengine/visualization/vis_backend.py
View File

@ -438,7 +438,7 @@ class WandbVisBackend(BaseVisBackend):
assert isinstance(self._init_kwargs, dict) assert isinstance(self._init_kwargs, dict)
allow_val_change = self._init_kwargs.get('allow_val_change', False) allow_val_change = self._init_kwargs.get('allow_val_change', False)
self._wandb.config.update( self._wandb.config.update(
dict(config), allow_val_change=allow_val_change) config.to_dict(), allow_val_change=allow_val_change)
self._wandb.run.log_code(name=self._log_code_name) self._wandb.run.log_code(name=self._log_code_name)
@force_init_env @force_init_env
@ -752,7 +752,7 @@ class MLflowVisBackend(BaseVisBackend):
""" """
self.cfg = config self.cfg = config
if self._tracked_config_keys is None: if self._tracked_config_keys is None:
self._mlflow.log_params(self._flatten(self.cfg)) self._mlflow.log_params(self._flatten(self.cfg.to_dict()))
else: else:
tracked_cfg = dict() tracked_cfg = dict()
for k in self._tracked_config_keys: for k in self._tracked_config_keys:
@ -915,7 +915,7 @@ class ClearMLVisBackend(BaseVisBackend):
config (Config): The Config object config (Config): The Config object
""" """
self.cfg = config self.cfg = config
self._task.connect_configuration(vars(config)) self._task.connect_configuration(config.to_dict())
@force_init_env @force_init_env
def add_image(self, def add_image(self,
@ -1216,7 +1216,7 @@ class DVCLiveVisBackend(BaseVisBackend):
""" """
assert isinstance(config, Config) assert isinstance(config, Config)
self.cfg = config self.cfg = config
self._dvclive.log_params(self._to_dvc_paramlike(self.cfg)) self._dvclive.log_params(self._to_dvc_paramlike(self.cfg.to_dict()))
@force_init_env @force_init_env
def add_image(self, def add_image(self,