need to update the config by
https://github.com/princeton-nlp/NLProofS
https://pytorch-lightning.readthedocs.io/en/1.6.0/common/lightning_cli.html
python main.py fit --config ruletaker.yaml
tensorboard --logdir lightning_logs/version_9