Spaces:
Sleeping
Sleeping
clear existing logger handler
Browse files
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 ''}"
|