Using load_dataset.set_transform() function along with Trainer class

[lewtun]

hey @tkon3, when you say “it fails” what do you mean exactly? there’s a related thread on set_transform here that might be useful: Understanding set_transform