scitex_ml.optim

Scitex optim module.

scitex_ml.optim.get(optim_str)

Get optimizer class by name.

DEPRECATED: Use get_optimizer instead.

scitex_ml.optim.get_optimizer(name)

Get optimizer class by name.

Parameters:

name (str) – Optimizer name (adam, ranger, rmsprop, sgd)

Returns:

Optimizer class

Raises:

ValueError – If optimizer name is not supported

scitex_ml.optim.set(models, optim_str, lr)

Sets an optimizer to models.

DEPRECATED: Use set_optimizer instead.

scitex_ml.optim.set_optimizer(models, optimizer_name, lr)

Set optimizer for models.

Parameters:

• models – Model or list of models

• optimizer_name (str) – Name of optimizer

• lr (float) – Learning rate

Returns:

Configured optimizer instance

Modules

Ranger_Deep_Learning_Optimizer