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()
}