You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I was testing the UNet model along with a mobilenetv2 and saw that there are a lot non-trainable params, which I could not explain. I quick lookup in the model revealed that besides the non-trainable BN params, there are also normal kernels that are marked as non-trainable. I guess this behavior is not desired.
Correct me if I am wrong, but I guess the default trainable parameter should be set to true ?
I was testing the UNet model along with a mobilenetv2 and saw that there are a lot non-trainable params, which I could not explain. I quick lookup in the model revealed that besides the non-trainable BN params, there are also normal kernels that are marked as non-trainable. I guess this behavior is not desired.
Correct me if I am wrong, but I guess the default
trainable
parameter should be set totrue
?TensorFlow-Advanced-Segmentation-Models/tensorflow_advanced_segmentation_models/models/_custom_layers_and_blocks.py
Lines 214 to 219 in c06d3d2
The text was updated successfully, but these errors were encountered: