cleaned the trainer so that we can direclty use a datasets with the top K logits. #2

Closed shamanez closed 1 month ago