import wandb in summary.py

Author: amaarora

timm/utils/summary.py

@@ -4,6 +4,7 @@
 """
 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
-            _logger.warning(f"Wandb project name not provided, defaulting to {args.model}")
+            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