File tree Expand file tree Collapse file tree 6 files changed +307
-298
lines changed
labml_nn/transformers/retro Expand file tree Collapse file tree 6 files changed +307
-298
lines changed Original file line number Diff line number Diff line change @@ -80,6 +80,7 @@ <h1>Retrieval-Enhanced Transformer (Retro)</h1>
8080< li > < a href ="model.html "> Model</ a > </ li >
8181< li > < a href ="dataset.html "> Dataset</ a > : Pre-calculate the nearest neighbors </ li >
8282< li > < a href ="train.html "> Training code</ a > </ li > </ ul >
83+ < p > < a href ="https://app.labml.ai/run/3113dd3ea1e711ec85ee295d18534021 "> < img alt ="View Run " src ="https://img.shields.io/badge/labml-experiment-brightgreen "> </ a > </ p >
8384
8485 </ div >
8586 < div class ='code '>
Load Diff Large diffs are not rendered by default.
Load Diff Large diffs are not rendered by default.
Original file line number Diff line number Diff line change 3232* [Model](model.html)
3333* [Dataset](dataset.html): Pre-calculate the nearest neighbors
3434* [Training code](train.html)
35+
36+ [](https://app.labml.ai/run/3113dd3ea1e711ec85ee295d18534021)
3537"""
Original file line number Diff line number Diff line change 99
1010This is the model definition for
1111 [RETRO](index.html).
12+
13+ [](https://app.labml.ai/run/3113dd3ea1e711ec85ee295d18534021)
1214"""
1315
1416import math
Original file line number Diff line number Diff line change 99
1010This is the training code for
1111 [RETRO](index.html).
12+
13+ [](https://app.labml.ai/run/3113dd3ea1e711ec85ee295d18534021)
1214"""
1315
1416import torch
You can’t perform that action at this time.
0 commit comments