hunk.code_window:     setup_default_logging()
       args, args_text = _parse_args()
       if args.log_wandb:
           wandb.init(project=args.experiment, config=args)
       args.prefetcher = not args.no_prefetcher
       args.distributed = False
       if 'WORLD_SIZE' in os.environ:
           args.distributed = int(os.environ['WORLD_SIZE']) > 1
       args.device = 'cuda:0'
  
hunk.labels: keep keep keep keep replace replace keep keep keep keep keep
hunk.after_edit:         if has_wandb:
               wandb.init(project=args.experiment, config=args)
           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`")
  
hunk.file_path: train.py
hunk.type: replace
file: torch>=1.4.0
  torchvision>=0.5.0
  pyyaml
  wandb