Hi @pvskand thanks for the quick reply! Can you elaborate, maybe with an example, on how to freeze the layers? If I don’t write a backward method, does that count as not changing the value of the layer’s weights?