[BUG] BERTClassifier drop_out rate에러

[Danden1]

🐛 Bug

네이버 리뷰코드에서 BERTClassifier의 dr_rate를 None이나 0.0으로 하면 오류가 발생합니다.

To Reproduce

또는

로 설정하면,

이와 같은 오류가 뜹니다.

if self.dr_rate:
        out = self.dropout(pooler)
else:
        out = pooler

return self.classifier(out)

로 고쳐야 할 것 같습니다.

[bage79]

@Danden1 오류로 인해, 불편을 드려서 죄송합니다.

말씀하신 내용을 반영하여 커밋하였습니다. https://github.com/SKTBrain/KoBERT/commit/e1f2f37055e7460d8427f6912579c0162cb69831

감사합니다. 😍