b99870ac55f3773ff2cd59fc298f9a4f987589da,src/main.py,,main,#Any#,46

Before Change

    log.info("Saved config to %s", config_file)

    git_branch_name = os.popen("git rev-parse --abbrev-ref HEAD").read().strip()
    git_sha = os.popen("git rev-parse HEAD").read().strip()
    log.info("On git branch {} at checkpoint {}.".format(git_branch_name, git_sha))

    seed = random.randint(1, 10000) if args.random_seed < 0 else args.random_seed
    random.seed(seed)

After Change

    config.write_params(args, config_file)
    log.info("Saved config to %s", config_file)

    try:
      log.info("Waiting on git info....")
      git_branch_name = subprocess.check_output("git rev-parse --abbrev-ref HEAD", stderr=subprocess.STDOUT, timeout=10, shell=True)
      git_sha = subprocess.check_output("git rev-parse HEAD", stderr=subprocess.STDOUT, timeout=10, shell=True)
      log.info("On git branch {} at checkpoint {}.".format(git_branch_name, git_sha))
    except subprocess.TimeoutExpired:
      git_branch_name.kill()
      log.warn("Git info not found. Moving right along...") 
      
    seed = random.randint(1, 10000) if args.random_seed < 0 else args.random_seed
    random.seed(seed)
    torch.manual_seed(seed)
    log.info("Using random seed %d", seed)

Instances