sharktide's picture
Create custom_objects.py
6bff2cd verified
Raw History Blame Contribute Delete
91 Bytes
import tensorflow as tf
CUSTOM_OBJECTS = {
"mse": tf.keras.losses.MeanSquaredError()
}