Fixed a key error in layer type dict that came up during the unpickling of SdA models.