This commit is contained in:
epwalsh
2019-12-30 14:15:26 -08:00
parent 0be9e5a7f0
commit b28c3fd04c

View File

@ -38,6 +38,7 @@ impl BPE {
}
}
/// Initialize a BPE model with [dropout](https://arxiv.org/abs/1910.13267).
pub fn with_dropout(
vocab: HashMap<String, u32>,
vocab_r: HashMap<u32, String>,