Browse Source

Update git_describe() for remote dir usage (#2606)

5.0
Glenn Jocher GitHub 3 years ago
parent
commit
77415a42e5
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
  1. +4
    -3
      utils/torch_utils.py

+ 4
- 3
utils/torch_utils.py View File

@@ -45,9 +45,10 @@ def init_torch_seeds(seed=0):

def git_describe():
# return human-readable git description, i.e. v5.0-5-g3e25f1e https://git-scm.com/docs/git-describe
if Path('.git').exists():
return subprocess.check_output('git describe --tags --long --always', shell=True).decode('utf-8')[:-1]
else:
s = f'git -C {Path(__file__).resolve().parent} describe --tags --long --always'
try:
return subprocess.check_output(s, shell=True).decode()[:-1]
except subprocess.CalledProcessError as e:
return ''



Loading…
Cancel
Save