I’m working on supporting automatic model detection/logging for PyTorch models for our Machine learning platform https://iko.ai 1 and I want to know what is the base class for models that only models inherit from? i.e. I’m looking for the class X that passes this condition: If a class Y inherits from class X, Y is a PyTorch model.
Is it torch.nn.modules.module.Module ?