RNNs with Darknet
Installing Darknet
- We need to install our modified version of Darknet:
- Darknet Installation
Using RNNs with Darknet
- Getting some calculated weights from several writers:
wget https://pjreddie.com/media/files/shakespeare.weights wget https://pjreddie.com/media/files/kant.weights wget https://pjreddie.com/media/files/tolstoy.weights
- Generating new text
./darknet rnn generate cfg/rnn.cfg shakespeare.weights -srand 0
- Parameters:
-srand N -> N determines the random seed
- Training with other corpus of text
- Download a text file. Some corpus of text online:
George Meredith http://www.gutenberg.org/ebooks/4500 Mark Twain http://www.gutenberg.org/ebooks/3200 More http://www.gutenberg.org/ebooks/search/?query=The+Complete+Works+of
- We train the neural network with the new text (This might take a while):
./darknet rnn train cfg/rnn.train.cfg -file filename.txt
- Generating new text
./darknet rnn generate cfg/rnn.cfg mark.weights -srand 0
- Optional: We can modify file cfg/rnn.train.cfg file (batch = 50 is the value we set by default, the original value was 256)