carate.optimizer package

Submodules

carate.optimizer.optimizer module

Module to handle optimizer initalization

@author Julian M. Kleber

carate.optimizer.optimizer.get_optimizer(optimizer_str: str, model_net: Model, learning_rate: float) → Optimizer

The get_optimizer function takes in a string and returns the corresponding optimizer.

Args:

optimizer_str (str): The name of the desired optimizer.

Returns:

Type[torch.optim.Optimizer]: The corresponding torch optimization function for the given string.

Parameters:

• optimizer_str:str – Used to Specify the type of optimizer we want to use.

• model – Used to Get the parameters of the model.

Returns:

The optimizer of the network.

Doc-author:

Trelent

Module contents