We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 20a1fa6 commit e1e4c9bCopy full SHA for e1e4c9b
train.py
@@ -55,7 +55,7 @@
55
try:
56
import wandb
57
has_wandb = True
58
-except ImportError:
+except ImportError:
59
has_wandb = False
60
61
torch.backends.cudnn.benchmark = True
@@ -326,14 +326,14 @@ def _parse_args():
326
def main():
327
setup_default_logging()
328
args, args_text = _parse_args()
329
-
+
330
if args.log_wandb:
331
if has_wandb:
332
wandb.init(project=args.experiment, config=args)
333
- else:
+ else:
334
_logger.warning("You've requested to log metrics to wandb but package not found. "
335
"Metrics not being logged to wandb, try `pip install wandb`")
336
337
args.prefetcher = not args.no_prefetcher
338
args.distributed = False
339
if 'WORLD_SIZE' in os.environ:
0 commit comments