jinaai

/

jina-bert-implementation

like 7

Follow

Jina AI 310

Can you move dropout_p to the constructor?

You mean instead of using self.dropout.p we use something like self.dropout_p that we set in the constructor? We will still need to check self.training in the forward pass.