mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 21:23:10 +08:00
[tune] Remove unnecessary wandb group parameter (#10950)
This commit is contained in:
committed by
Barak Michener
parent
f371eb334c
commit
13d703be12
@@ -138,7 +138,6 @@ def wandb_mixin(func):
|
||||
|
||||
"""
|
||||
func.__mixins__ = (WandbTrainableMixin, )
|
||||
func.__wandb_group__ = func.__name__
|
||||
return func
|
||||
|
||||
|
||||
@@ -342,8 +341,8 @@ class WandbLogger(Logger):
|
||||
exclude_results += ["config"]
|
||||
|
||||
# Fill trial ID and name
|
||||
trial_id = self.trial.trial_id
|
||||
trial_name = str(self.trial)
|
||||
trial_id = self.trial.trial_id if self.trial else None
|
||||
trial_name = str(self.trial) if self.trial else None
|
||||
|
||||
# Project name for Wandb
|
||||
try:
|
||||
@@ -353,7 +352,8 @@ class WandbLogger(Logger):
|
||||
"You need to specify a `project` in your wandb `config` dict.")
|
||||
|
||||
# Grouping
|
||||
wandb_group = wandb_config.pop("group", self.trial.trainable_name)
|
||||
wandb_group = wandb_config.pop(
|
||||
"group", self.trial.trainable_name if self.trial else None)
|
||||
|
||||
# remove unpickleable items!
|
||||
config = _clean_log(config)
|
||||
|
||||
Reference in New Issue
Block a user