-
class
Builder Bases:
abc.ABCBuilder builds TF1 computation graph to be used for training.
This class defines require methods for Builder implementations.
-
abstract
build() This method builds the computation graph for training.
Returns:
-
abstract
-
class
TFBuilder(task, initial_learning_rate, train_data_source, model, loss=None, optimizer=None, multi_gpu=False, aux_ops=None, val_data_source=None, dynamic_input_shape=False, extra_inputs=None) Bases:
ai4med.workflows.builders.builder.Builder-
build() This method builds the computation graph for training.
Returns:
-