bugfix resunet-a

yingkaisha · Jun 13, 2021 · 2383db7 · 2383db7
1 parent 75f717d
commit 2383db7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_unet_collection/_model_resunet_a_2d.py b/keras_unet_collection/_model_resunet_a_2d.py
@@ -282,7 +282,7 @@ def resunet_a_2d(input_size, filter_num, dilation_num, n_labels,
 
     OUT = CONV_output(X, n_labels, kernel_size=1, activation=output_activation, name='{}_output'.format(name))
 
-    model = Model([IN], [OUT,])
+    model = Model([IN], [OUT,], name='{}_model'.format(name))
 
     return model