Why does PretrainedLayer in particular need a specialized class?