Seq2seq Couplet
| Project Name | Stars | Downloads | Repos Using This | Packages Using This | Most Recent Commit | Total Releases | Latest Release | Open Issues | License | Language |
|---|---|---|---|---|---|---|---|---|---|---|
| Tensorflow | 172,632 | 327 | 77 | 2 hours ago | 46 | October 23, 2019 | 2,246 | apache-2.0 | C++ | |
| An Open Source Machine Learning Framework for Everyone | ||||||||||
| Transformers | 89,017 |  | 64 | 911 | 3 hours ago | 91 | June 21, 2022 | 634 | apache-2.0 | Python |
| 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX. | ||||||||||
| Pytorch | 64,630 | 146 | 2 hours ago | 23 | August 10, 2022 | 11,508 | other | C++ | ||
| Tensors and Dynamic neural networks in Python with strong GPU acceleration | ||||||||||
| Keras | 57,766 |  | 330 | 15 hours ago | 68 | May 13, 2022 | 375 | apache-2.0 | Python | |
| Deep Learning for humans | ||||||||||
| Cs Video Courses | 53,706 | 11 days ago | 14 | |||||||
| List of Computer Science courses with video lectures. | ||||||||||
| Faceswap | 43,824 | a month ago | 15 | gpl-3.0 | Python | |||||
| Deepfakes Software For All | ||||||||||
| Tensorflow Examples | 42,312 | 5 months ago | 218 | other | Jupyter Notebook | |||||
| TensorFlow Tutorial and Examples for Beginners (support TF v1 & v2) | ||||||||||
| D2l Zh | 41,168 | 1 | 12 hours ago | 45 | March 25, 2022 | 24 | apache-2.0 | Python | ||
| 《动手学深度学习》：面向中文读者、能运行、可讨论。中英文版被60多个国家的400多所大学用于教学。 | ||||||||||
| 100 Days Of Ml Code | 40,344 | 18 days ago | 61 | mit | ||||||
| 100 Days of ML Coding | ||||||||||
| Deepfacelab | 37,877 | 19 days ago | 510 | gpl-3.0 | Python | |||||
| DeepFaceLab is the leading software for creating deepfakes. | ||||||||||
This is a project use seq2seq model to play couplets () This project is written with Tensorflow. You can try the demo at https://ai.binwang.me/couplet.
Pre-requirements
- Tensorflow
- Python 3.6
- Dataset
Dataset
You will need some data to run this program, the dataset can be downloaded from this project.
** Note: If you are using your own dataset, you need to add <s> and <\s> as the first two line into the vocabs file. **
Usage
Train
Open couplet.py and config the file locations and hyperparams. Then run python couplet.py to train the model. You can see the training loss and bleu score at Tensorbloard. You may need to re-config learning_rate when you find the loss stops descresing. Here is an example of the loss graph:
If you stoped the training and want to continue to train it. You can set restore_model to True and use m.train(<epoches>, start=<start>), which start is the steps you've already run.
I've trained the model on a Nvidia GTX-1080 GPU for about 4 days.
Run the trained model
Open server.py and config the vocab_file and model_dir params. Then run python server.py will start a web service that can play couplet.
Or build the Docker image with Dockerfile and run it with Docker. Remember to mount the correct model file paths into the Docker container.
Examples
Here are some examples generated by this model: