fffiloni commited on
Commit
0572ad4
1 Parent(s): d61b629

clear existing logger handler

Browse files
Files changed (1) hide show
  1. main.py +2 -0
main.py CHANGED
@@ -19,7 +19,9 @@ def setup(args):
19
  seed_everything(args.seed)
20
  bf.makedirs(f"{args.save_dir}/logs/{args.task}")
21
  # Set up logging and name settings
 
22
  logger = logging.getLogger()
 
23
  settings = (
24
  f"{args.model}{'_' + args.prompt if args.task == 't2i-compbench' else ''}"
25
  f"{'_no-optim' if args.no_optim else ''}_{args.seed if args.task != 'geneval' else ''}"
 
19
  seed_everything(args.seed)
20
  bf.makedirs(f"{args.save_dir}/logs/{args.task}")
21
  # Set up logging and name settings
22
+ # Get the root logger and clear existing handlers
23
  logger = logging.getLogger()
24
+ logger.handlers.clear() # Clear existing handlers
25
  settings = (
26
  f"{args.model}{'_' + args.prompt if args.task == 't2i-compbench' else ''}"
27
  f"{'_no-optim' if args.no_optim else ''}_{args.seed if args.task != 'geneval' else ''}"