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: from .byobnet import *
  from .cspnet import *
  from .densenet import *
  from .dla import *
  from .dpn import *
  from .efficientnet import *
  from .gluon_resnet import *
  from .gluon_xception import *
  from .hardcorenas import *
  from .hrnet import *
  from .inception_resnet_v2 import *
  from .inception_v3 import *
  from .inception_v4 import *
  from .mobilenetv3 import *
  from .nasnet import *
  from .nfnet import *
  from .pit import *
  from .pnasnet import *
  from .regnet import *
  from .res2net import *
  from .resnest import *
  from .resnet import *
  from .resnetv2 import *
  from .rexnet import *
  from .selecsls import *
  from .senet import *
  from .sknet import *
  from .tnt import *
  from .tresnet import *
  from .vgg import *
  from .vision_transformer import *
  from .vision_transformer_hybrid import *
  from .vovnet import *
  from .xception import *
  from .xception_aligned import *
  
  from .factory import create_model, split_model_name, safe_model_name
  from .helpers import load_checkpoint, resume_checkpoint, model_parameters
  from .layers import TestTimePoolHead, apply_test_time_pool
  from .layers import convert_splitbn_model
  from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable, is_no_jit, set_no_jit
  from .registry import *
  
  