From e1e4c9bbae292ee983f0e606283f66ee0598b1d4 Mon Sep 17 00:00:00 2001 From: Jakub Kaczmarzyk Date: Wed, 18 May 2022 10:17:02 -0400 Subject: [PATCH] rm whitespace --- train.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/train.py b/train.py index 6f31e295..b6aade12 100755 --- a/train.py +++ b/train.py @@ -55,7 +55,7 @@ except AttributeError: try: import wandb has_wandb = True -except ImportError: +except ImportError: has_wandb = False torch.backends.cudnn.benchmark = True @@ -326,14 +326,14 @@ def _parse_args(): def main(): setup_default_logging() args, args_text = _parse_args() - + if args.log_wandb: if has_wandb: wandb.init(project=args.experiment, config=args) - else: + else: _logger.warning("You've requested to log metrics to wandb but package not found. " "Metrics not being logged to wandb, try `pip install wandb`") - + args.prefetcher = not args.no_prefetcher args.distributed = False if 'WORLD_SIZE' in os.environ: