Getting-Things-Done-with-Pytorch icon indicating copy to clipboard operation
Getting-Things-Done-with-Pytorch copied to clipboard

fix "bert_model" outputs

Open spacewaterbear opened this issue 3 years ago • 0 comments

(last_hidden_state, pooled_output) othervise last_hidden_state.shape return an AttributeError: 'str' object has no attribute 'shape'

spacewaterbear avatar Apr 03 '21 16:04 spacewaterbear