use of org.apache.ignite.ml.environment.parallelism.DefaultParallelismStrategy in project ignite by apache.
the class LearningEnvironmentBuilderTest method withParallelismStrategy.
/**
*/
@Test
public void withParallelismStrategy() {
assertTrue(LearningEnvironmentBuilder.defaultBuilder().withParallelismStrategyDependency(part -> NoParallelismStrategy.INSTANCE).buildForTrainer().parallelismStrategy() instanceof NoParallelismStrategy);
assertTrue(LearningEnvironmentBuilder.defaultBuilder().withParallelismStrategyDependency(part -> new DefaultParallelismStrategy()).buildForTrainer().parallelismStrategy() instanceof DefaultParallelismStrategy);
}
Aggregations