1 write to ParallelTraining
Microsoft.ML.LightGbm (1)
LightGbmTrainerBase.cs (1)
379
ParallelTraining
= LightGbmTrainerOptions.ParallelTrainer != null ? LightGbmTrainerOptions.ParallelTrainer.CreateComponent(Host) : new SingleTrainer();
13 references to ParallelTraining
Microsoft.ML.LightGbm (13)
LightGbmTrainerBase.cs (13)
494
if (
ParallelTraining
.ParallelType() != "serial" &&
ParallelTraining
.NumMachines() > 1)
496
GbmOptions["tree_learner"] =
ParallelTraining
.ParallelType();
497
var otherParams =
ParallelTraining
.AdditionalParams();
504
Contracts.CheckValue(
ParallelTraining
.GetReduceScatterFunction(), nameof(
ParallelTraining
.GetReduceScatterFunction));
505
Contracts.CheckValue(
ParallelTraining
.GetAllgatherFunction(), nameof(
ParallelTraining
.GetAllgatherFunction));
507
ParallelTraining
.NumMachines(),
508
ParallelTraining
.Rank(),
509
ParallelTraining
.GetReduceScatterFunction(),
510
ParallelTraining
.GetAllgatherFunction()
517
if (
ParallelTraining
.NumMachines() > 1)