1 write to ParallelTraining
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
355
ParallelTraining
= LightGbmTrainerOptions.ParallelTrainer != null ? LightGbmTrainerOptions.ParallelTrainer.CreateComponent(Host) : new SingleTrainer();
13 references to ParallelTraining
Microsoft.ML.LightGbm (13)
LightGbmTrainerBase.cs (13)
470
if (
ParallelTraining
.ParallelType() != "serial" &&
ParallelTraining
.NumMachines() > 1)
472
GbmOptions["tree_learner"] =
ParallelTraining
.ParallelType();
473
var otherParams =
ParallelTraining
.AdditionalParams();
480
Contracts.CheckValue(
ParallelTraining
.GetReduceScatterFunction(), nameof(
ParallelTraining
.GetReduceScatterFunction));
481
Contracts.CheckValue(
ParallelTraining
.GetAllgatherFunction(), nameof(
ParallelTraining
.GetAllgatherFunction));
483
ParallelTraining
.NumMachines(),
484
ParallelTraining
.Rank(),
485
ParallelTraining
.GetReduceScatterFunction(),
486
ParallelTraining
.GetAllgatherFunction()
493
if (
ParallelTraining
.NumMachines() > 1)