I use PyTorch Lightning (which I would personally recommend over Poutyne etc.)
You need to be more specific when you say ‘does not work’ : what is the error? Share a colab/notebook link and I will be able to debug.