custom_objects.py

import tensorflow as tf

CUSTOM_OBJECTS = {
    "mse": tf.keras.losses.MeanSquaredError()
}