updated experimental simpleKnnLoss