warn_if_parameter_server_with_multi_gpu: Warn the user that training will not fully leverage all the...

View source: R/fw_utils.R

warn_if_parameter_server_with_multi_gpuR Documentation

Warn the user that training will not fully leverage all the GPU cores

Description

Warn the user that training will not fully leverage all the GPU cores if parameter server is enabled and a multi-GPU instance is selected. Distributed training with the default parameter server setup doesn't support multi-GPU instances.

Usage

warn_if_parameter_server_with_multi_gpu(training_instance_type, distribution)

Arguments

training_instance_type

(str): A string representing the type of training instance selected.

distribution

(dict): A dictionary with information to enable distributed training. (Defaults to None if distributed training is not enabled.).


DyfanJones/sagemaker-r-local documentation built on June 14, 2022, 10:32 p.m.