bitorch.models.register_custom_model

bitorch.models.register_custom_model(custom_model: Type[Model]) → None

Register a custom (external) model in bitorch.

Parameters:

custom_model – the custom model which should be added to bitorch