Thomas Chaigneau

Any hint since March ? I'm looking for something similar. If it still doesn't exist, is there a chance to contribute for that feature ?

Hi @creativesalam, I stopped using Tensorflow in favor of PyTorch and PyTorch Lightning, which better suit my needs.