AIFEBaseModel function

Base class for objects using a pytorch model as core model.