import sys
from .kerasmodel import KerasModel,train_model
from .summary import summary
try:
    from .lightmodel import LightModel 
except Exception:
    print("torchkeras.LightModel can't be used!",file = sys.stderr)
__version__='3.3.1'
