minor change in main
This commit is contained in:
parent
f6d61f02f6
commit
9df9e07f9b
@ -264,6 +264,10 @@ class BoostLearnTask {
|
|||||||
|
|
||||||
int main(int argc, char *argv[]){
|
int main(int argc, char *argv[]){
|
||||||
xgboost::sync::Init(argc, argv);
|
xgboost::sync::Init(argc, argv);
|
||||||
|
if (xgboost::sync::IsDistributed()) {
|
||||||
|
std::string pname = xgboost::sync::GetProcessorName();
|
||||||
|
printf("start %s:%d\n", pname.c_str(), xgboost::sync::GetRank());
|
||||||
|
}
|
||||||
xgboost::random::Seed(0);
|
xgboost::random::Seed(0);
|
||||||
xgboost::BoostLearnTask tsk;
|
xgboost::BoostLearnTask tsk;
|
||||||
int ret = tsk.Run(argc, argv);
|
int ret = tsk.Run(argc, argv);
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user