search

huggingface / candle

Minimalist ML framework for Rust
Apache License 2.0
15.88k stars 962 forks source link

Implement BertPooling #2601

Open DimitriTimoz opened 2 weeks ago

DimitriTimoz commented 2 weeks ago

In the implementation of the BertModel, the BertPooler was not implemented. Here is the implementation of the BertPooler as it was done with python.