Migrate transformertf to LightningCLI for easier training. Arguments are linked together in the transformertf.main.py