[jira] [Created] (MADLIB-1394) DL: Predict changes for asymmetric cluster config

["Ekta Khanna (Jira)" <ji...@apache.org>]

Ekta Khanna created MADLIB-1394:
-----------------------------------

             Summary: DL: Predict changes for asymmetric cluster config
                 Key: MADLIB-1394
                 URL: https://issues.apache.org/jira/browse/MADLIB-1394
             Project: Apache MADlib
          Issue Type: New Feature
            Reporter: Ekta Khanna
             Fix For: v1.17


*Predict*
For predict we only support symmetric clusters, either all CPU or with GPUs on all hosts.
{code}
madlib_keras_predict(
    model_table,
    test_table,
    id_col,
    independent_varname,
    output_table,
    pred_type,
    use_gpus  -- changed definition
    )
{code}
{use_gpus} (optional)
BOOLEAN, *default*: FALSE (i.e., CPU). Determines whether GPUs are to be used for prediction/inference.  Set to TRUE to use GPUs.

*Note*
The prediction function uses the whole cluster.  So if you are using GPUs, it requires that GPUs are attached to all hosts.  This is different from the fit() and evaluate() functions that support GPUs on only some of the hosts (asymmetric cluster).  Therefore, if you have GPUs only on some of the hosts, set this parameter to FALSE to use CPUs.

*Predict BYOM*
Same as predict above ^^^.



--
This message was sent by Atlassian Jira
(v8.3.4#803005)