def __init__(
self,
api_key: Optional[str] = None,
save_dir: Optional[str] = None,
workspace: Optional[str] = None,
project_name: Optional[str] = None,
rest_api_key: Optional[str] = None,
experiment_name: Optional[str] = None,
experiment_key: Optional[str] = None,
offline: bool = False,
**kwargs,
):
if not _COMET_AVAILABLE:
raise ImportError(
"You want to use `comet_ml` logger which is not installed yet,"
" install it with `pip install comet-ml`.")
super().__init__()
self._experiment = None
# Determine online or offline mode based on which arguments were passed to CometLogger
api_key = api_key or get_api_key(None, get_config())
if api_key is not None and save_dir is not None:
self.mode = "offline" if offline else "online"
self.api_key = api_key
self._save_dir = save_dir
elif api_key is not None:
self.mode = "online"
self.api_key = api_key
self._save_dir = None
elif save_dir is not None:
self.mode = "offline"
self._save_dir = save_dir
else:
# If neither api_key nor save_dir are passed as arguments, raise an exception
raise MisconfigurationException(
"CometLogger requires either api_key or save_dir during initialization."
)
log.info(f"CometLogger will be initialized in {self.mode} mode")
self.workspace = workspace
self._project_name = project_name
self._experiment_key = experiment_key
self._experiment_name = experiment_name
self._kwargs = kwargs
self._future_experiment_key = None
if rest_api_key is not None:
# Comet.ml rest API, used to determine version number
self.rest_api_key = rest_api_key
self.comet_api = API(self.rest_api_key)
else:
self.rest_api_key = None
self.comet_api = None
self._kwargs = kwargs
def get_api_key_or_login(self, api_key):
# ok
api_key = get_api_key(api_key, self.config)
if api_key is None:
return self.create_or_login()
Reporting.report("mlflow_existing_user", api_key=api_key)
return (api_key, None)
"""
config_path = Path(config_path)
model = CenterTrack(
model=model_type,
config_path=config_path,
epoch_size=epoch_size,
batch_size=batch_size,
n_workers=n_workers,
img_scale=img_scale,
label_scale=label_scale,
grayscale=grayscale,
)
print(f"# parameters: {sum(p.numel() for p in model.parameters()):,}")
api_key = get_api_key(None, get_config())
project_name = "vsrl"
exp_name = f"detector_{config_path.name.split('.')[0]}"
if use_logger:
logger = CometLogger(
api_key,
save_dir,
project_name=project_name,
experiment_name=exp_name,
force_offline=force_offline,
log_env_gpu=False,
log_env_cpu=False,
)
logger.log_hyperparams({
"img_scale": img_scale,