Hi @matheo_r  thanks for reply… actually I don’t know ‘element 0 of tensors’ refers to what!!!
if it means the first layer of my network, its the freezed one…
this is my network:
conv1 freezed
conv2 freezed
conv3 freezed
conv4 trainable
conv5 trainable
conv6 trainable
fully connected freezed
fully connected freezed