Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import torch | |
import triton | |
import triton.language as tl | |
from itertools import product | |
@triton.jit | |
def copy_kernel( | |
output_ptr, input_ptr, |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
"""Character based language modeling with multi-layer GRUs. | |
To train the model: | |
python3 tf_char_rnn.py --mode training \ | |
--logdir path/to/logdir --corpus path/to/corpus.txt | |
To generate text from seed words: | |
python3 tf_char_rnn.py --mode sampling \ |