lfcnn.training package

Subpackages

Module contents

The LFCNN training module.

The training module defines some (optional) keras.Model subclasses implementing specific training strategies, mostly in the context of multi-task learning.

The keras.Model subclasses can then be passed to lfcnn.Model instantiation.

lfcnn.training.get_model_cls(model_cls)[source]

Given a model class name, returns the corresponding model class.

Parameters

model_cls (str) – Name of the model class.

Return type

Optional[Callable]

Returns

Model class.