From 3f028ebc0f8cfeb1bb70d821ff3fdcc1a2f64173 Mon Sep 17 00:00:00 2001 From: Aman Arora Date: Thu, 8 Apr 2021 03:48:51 -0400 Subject: [PATCH] import wandb in summary.py --- timm/utils/summary.py | 1 + train.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/timm/utils/summary.py b/timm/utils/summary.py index 44a89afb..10e317c5 100644 --- a/timm/utils/summary.py +++ b/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 diff --git a/train.py b/train.py index 7c6f9d4b..172008a2 100755 --- a/train.py +++ b/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