def save_model()

in train/train.py [0:0]


def save_model(model,outputpath):
    print("saving model in path - ",outputpath)
    tf.saved_model.save(model,os.path.join(outputpath ,'1'))