Default project name to None
parent
8db8ff346f
commit
f8bb13f640
3
train.py
3
train.py
|
@ -301,9 +301,6 @@ def main():
|
|||
args, args_text = _parse_args()
|
||||
|
||||
if args.use_wandb:
|
||||
if not args.wandb_project_name:
|
||||
args.wandb_project_name = f'timm_{args.model}'
|
||||
_logger.warning(f"Wandb project name not provided, defaulting to timm_{args.model}")
|
||||
wandb.init(project=args.wandb_project_name, config=args)
|
||||
|
||||
args.prefetcher = not args.no_prefetcher
|
||||
|
|
Loading…
Reference in New Issue