<reponame>yaoyue123/ai_learn
from poetry_model import PoetryModel


if __name__ == '__main__':
    poetry = PoetryModel()
    poetry.train(epoch=20)

from poetry_model import PoetryModel


if __name__ == '__main__':
    poetry = PoetryModel()
    poetry.train(epoch=20)