This version doesn't freeze the weights maps.