when I use bert-large-uncased to load bert,runtimeError occured,but base-uncase is ok
using BertModel.from_pretrained( path of bert-large-uncased) caused error

RuntimeError: $ Torch: invalid memory size -- maybe an overflow? at ..\aten\src\TH\THGeneral.cpp:188

But using  BertModel.from_pretrained( path of bert-case-uncased)  can work

Yes, probably overflow. Try a smaller batch size?<|||||>This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.
