Getting-Things-Done-with-Pytorch
Getting-Things-Done-with-Pytorch copied to clipboard
fix "bert_model" outputs
(last_hidden_state, pooled_output) othervise last_hidden_state.shape return an AttributeError: 'str' object has no attribute 'shape'