eval_retrieval_feature_extract.py [124:131]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    torch.cuda.set_device(args.local_rank)
    torch.distributed.init_process_group(backend='nccl', init_method='env://')
    cudnn.benchmark = True

    os.makedirs(args.output_dir, exist_ok=True)
    logger = setup_logger(output=args.output_dir, distributed_rank=dist.get_rank(), name="vclr")
    if dist.get_rank() == 0:
        path = os.path.join(args.output_dir, "config.json")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train_vclr.py [179:186]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    torch.cuda.set_device(args.local_rank)
    torch.distributed.init_process_group(backend='nccl', init_method='env://')
    cudnn.benchmark = True

    os.makedirs(args.output_dir, exist_ok=True)
    logger = setup_logger(output=args.output_dir, distributed_rank=dist.get_rank(), name="vclr")
    if dist.get_rank() == 0:
        path = os.path.join(args.output_dir, "config.json")
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -