初始化项目,由ModelHub XC社区提供模型
Model: ayh015/myLightningOPD Source: Original Platform
This commit is contained in:
24
slime/utils/tracking_utils.py
Normal file
24
slime/utils/tracking_utils.py
Normal file
@@ -0,0 +1,24 @@
|
||||
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
import wandb
|
||||
from slime.utils.tensorboard_utils import _TensorboardAdapter
|
||||
|
||||
from . import wandb_utils
|
||||
|
||||
|
||||
def init_tracking(args, primary: bool = True, **kwargs):
|
||||
if primary:
|
||||
wandb_utils.init_wandb_primary(args, **kwargs)
|
||||
else:
|
||||
wandb_utils.init_wandb_secondary(args, **kwargs)
|
||||
|
||||
|
||||
# TODO further refactor, e.g. put TensorBoard init to the "init" part
|
||||
def log(args, metrics, step_key: str):
|
||||
if args.use_wandb:
|
||||
wandb.log(metrics)
|
||||
|
||||
if args.use_tensorboard:
|
||||
metrics_except_step = {k: v for k, v in metrics.items() if k != step_key}
|
||||
_TensorboardAdapter(args).log(data=metrics_except_step, step=metrics[step_key])
|
||||
Reference in New Issue
Block a user