dlr: fixup

This commit is contained in:
Starbeamrainbowlabs 2023-03-03 22:09:05 +00:00
parent 5c6789bf40
commit e9dcbe3863
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -163,10 +163,6 @@ else:
model = tf.keras.models.load_model(PATH_CHECKPOINT, custom_objects={
# Tell Tensorflow about our custom layers so that it can deserialise models that use them
"LossCrossEntropyDice": LossCrossEntropyDice,
"MetricDice": MetricDice,
"MetricSensitivity": MetricSensitivity,
"MetricSpecificity": MetricSpecificity
"MetricMeanIoU": MetricMeanIoU
})