In order to work, is still needed a pytorch installation. Any plans to update the utils.py file to tensorflow?