neodroidvision.utilities.torch_utilities.output_activation.output_activation_module

Created on 03/06/2020

Classes

OutputActivationModule(model, output_activation)

For adding output activation to traced torchscript architectures