brainspy.utils.manager#

Class to retrieve data specified in a dictionary used to train a model. The class methods require a configuration dictionary with data keys specific to do a task. It can be used to get an optimizer,fitness function,driver or an algorithm for training a model.

Functions

get_adam(model[, configs])

To get an Adam optimizer object which include added information to train a specific model.

get_algorithm(name)

To get a default train function for either GA - genetic algorithm or GD - Gradient Descent, based on its name.

get_criterion(name)

Returns a loss/fitness function from brainspy.algorithms.modules.signal module given a string name.

get_driver(configs)

To get an instance of a driver object from brainspy.processors.hardware.drivers.nidaq/cdaq based on a configurations dictionary.

get_optimizer(model, configs)

Gets either a genetic algorithm or a gradient descent pytorch optimizer object from a dictionary.