CNN trainable layers

[vdplasthijs]

Ensure that CNN layers are correctly labelled as trainable modules or not. (Depending on init strategy pretrained/random).

(Do this before training CNNs)