tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-02-12 13:39:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

ddp check

Browse Source
This commit is contained in:
geronimi73 2025-10-15 04:42:46 +02:00
parent 67aaca98f5
commit 41bd7b16f2
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nanochat/common.py
View File

@ -76,8 +76,7 @@ def print_banner():
print0(banner)
def is_ddp():
# TODO is there a proper way
return int(os.environ.get('RANK', -1)) != -1
return dist.is_available() and dist.is_initialized()
def get_dist_info():
if is_ddp():