@@ -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 :
0 commit comments