import wandb in summary.py

timm/utils/summary.py

@@ -4,6 +4,7 @@ Hacked together by / Copyright 2020 Ross Wightman
 """
 import csv
 import os
+import wandb
 from collections import OrderedDict
 
 

train.py

@@ -302,8 +302,8 @@ def main():
     
     if args.use_wandb:
         if not args.wandb_project_name: 
-            args.wandb_project_name = args.model
+            args.wandb_project_name = f'timm_{args.model}'
-            _logger.warning(f"Wandb project name not provided, defaulting to {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