Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update rabit_tracker.py

Browse Source
This commit is contained in:
Tianqi Chen 2014-12-08 08:36:55 -08:00
parent b38fa40fa6
commit 39f2dcdfef
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/rabit_tracker.py
View File

@ -211,9 +211,9 @@ class Tracker:
shutdown[s.rank] = s shutdown[s.rank] = s
continue continue
assert s.cmd == 'start' or s.cmd == 'recover' assert s.cmd == 'start' or s.cmd == 'recover'
if s.cmd == 'recover': if s.cmd == 'recover'
assert s.rank >= 0 assert s.rank >= 0
print 'Recieve recover signal from %d' % s.rank print 'Recieve %s signal from %d' % (s.cmd, s.rank)
rank = s.decide_rank(job_map) rank = s.decide_rank(job_map)
if rank == -1: if rank == -1:
assert len(todo_nodes) != 0 assert len(todo_nodes) != 0