Skip to content

Commit c2ffa36

Browse files
authored
Update README.md
1 parent 152b632 commit c2ffa36

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,4 +19,5 @@ Furthermore write a function to convert the network output of vector encodings b
1919
[1] Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, Lukasz Kaiser, Illia Polosukhin:
2020
Attention is All you Need. NIPS 2017: 5998-6008
2121

22-
Once you have implemented and tested your version of attention run `sbatch scripts/train.slurm` to train your model on Bender. Once converged you can generate poetry via `sbatch scripts/generate.slurm`.
22+
Once you have implemented and tested your version of attention run `sbatch scripts/train.slurm` to train your model on Bender. Once converged you can generate poetry via `sbatch scripts/generate.slurm`.
23+
Run `src/model_chat.py` to talk to your model.

0 commit comments

Comments
 (0)