tomaat.frameworks package

Submodules

tomaat.frameworks.pytorch module

class tomaat.frameworks.pytorch.Prediction(model_path, input_arg_names, input_fields, output_fields, with_gpu=True)[source]

Bases: object

tomaat.frameworks.tf module

class tomaat.frameworks.tf.Prediction(model_path, input_tensors_names, input_fields, output_tensors_names, output_fields)[source]

Bases: object

Module contents