Changed NORB loading to float.
Added bkg_code to KMeansPartsLayer as well.