load_model#

load_model(device: Union[str, device] = 'cpu') Module[source]#

Load the pre-calculated prediction model and return it.