from tensorflow.keras.applications.resnet50 import ResNet50
from tensorflow.keras.optimizers import SGD


def get_model():
    model = ResNet50(weights=None)
    optimizer = SGD()
    model.compile(loss='categorical_crossentropy',
                  optimizer=optimizer,
                  metrics=["accuracy"])
    return model


Net = get_model()

from tensorflow.keras.applications.resnet50 import ResNet50
from tensorflow.keras.optimizers import SGD


def get_kernel(parsm, unparsed_args):
    model = ResNet50(weights=None)
    optimizer = SGD()
    model.compile(loss='categorical_crossentropy',
                  optimizer=optimizer,
                  metrics=["accuracy"])
    return model

fix tf