import wandb in summary.py

pull/550/head
Aman Arora 4 years ago
parent a9e5d9e5ad
commit 3f028ebc0f

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

@ -302,8 +302,8 @@ def main():
if args.use_wandb: if args.use_wandb:
if not args.wandb_project_name: 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) wandb.init(project=args.wandb_project_name, config=args)
args.prefetcher = not args.no_prefetcher args.prefetcher = not args.no_prefetcher

Loading…
Cancel
Save