Skip to content

Commit ec40334

Browse files
qgallouedecSunMarc
andauthored
Don't set run_name when none (#39695)
* Don't set run_name when none * revert --------- Co-authored-by: Marc Sun <[email protected]>
1 parent 551a89a commit ec40334

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

src/transformers/integrations/integration_utils.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -856,7 +856,7 @@ def setup(self, args, state, model, **kwargs):
856856
init_args = {}
857857
if trial_name is not None:
858858
init_args["name"] = trial_name
859-
init_args["group"] = args.run_name
859+
init_args["group"] = args.run_name or args.output_dir
860860
elif args.run_name is not None:
861861
init_args["name"] = args.run_name
862862
if args.run_name == args.output_dir:
@@ -1230,11 +1230,7 @@ def setup(self, args, state, model):
12301230

12311231
import comet_ml
12321232

1233-
# Do not use the default run_name as the experiment name
1234-
if args.run_name is not None and args.run_name != args.output_dir:
1235-
experiment_config = comet_ml.ExperimentConfig(name=args.run_name)
1236-
else:
1237-
experiment_config = comet_ml.ExperimentConfig()
1233+
experiment_config = comet_ml.ExperimentConfig(name=args.run_name)
12381234

12391235
self._experiment = comet_ml.start(online=online, mode=mode, experiment_config=experiment_config)
12401236
self._experiment.__internal_api__set_model_graph__(model, framework="transformers")
@@ -2357,10 +2353,12 @@ def setup(self, args, state, model, **kwargs):
23572353
combined_dict = {**{"peft_config": peft_config}, **combined_dict}
23582354
trial_name = state.trial_name
23592355
init_args = {}
2360-
if trial_name is not None:
2356+
if trial_name is not None and args.run_name is not None:
23612357
init_args["experiment_name"] = f"{args.run_name}-{trial_name}"
23622358
elif args.run_name is not None:
23632359
init_args["experiment_name"] = args.run_name
2360+
elif trial_name is not None:
2361+
init_args["experiment_name"] = trial_name
23642362
init_args["project"] = os.getenv("SWANLAB_PROJECT", None)
23652363

23662364
if self._swanlab.get_run() is None:

src/transformers/training_args.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1707,8 +1707,6 @@ def __post_init__(self):
17071707
self.metric_for_best_model = "loss"
17081708
if self.greater_is_better is None and self.metric_for_best_model is not None:
17091709
self.greater_is_better = not self.metric_for_best_model.endswith("loss")
1710-
if self.run_name is None:
1711-
self.run_name = self.output_dir
17121710
if self.framework == "pt" and is_torch_available():
17131711
if self.fp16_backend and self.fp16_backend != "auto":
17141712
warnings.warn(

0 commit comments

Comments
 (0)