Skip to content
Snippets Groups Projects
Commit 1fea8025 authored by leufen1's avatar leufen1
Browse files

changed default loss

parent 57511d09
Branches
Tags
5 merge requests!430update recent developments,!413update release branch,!412Resolve "release v2.0.0",!406Lukas issue368 feat prepare cnn class for filter benchmarking,!403Resolve "prepare CNN class for filter benchmarking"
Pipeline #93704 passed
......@@ -112,7 +112,8 @@ class CNN(AbstractModelClass): # pragma: no cover
# apply to model
self.set_model()
self.set_compile_options()
self.set_custom_objects(loss=custom_loss([keras.losses.mean_squared_error, var_loss]), var_loss=var_loss)
# self.set_custom_objects(loss=custom_loss([keras.losses.mean_squared_error, var_loss]), var_loss=var_loss)
self.set_custom_objects(loss=self.compile_options["loss"][0], var_loss=var_loss)
def _set_pooling(self, pooling):
try:
......@@ -221,7 +222,9 @@ class CNN(AbstractModelClass): # pragma: no cover
print(self.model.summary())
def set_compile_options(self):
self.compile_options = {"loss": [custom_loss([keras.losses.mean_squared_error, var_loss])],
# self.compile_options = {"loss": [custom_loss([keras.losses.mean_squared_error, var_loss])],
# "metrics": ["mse", "mae", var_loss]}
self.compile_options = {"loss": [keras.losses.mean_squared_error],
"metrics": ["mse", "mae", var_loss]}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment