search

SAI990323 / TALLRec

Apache License 2.0
213 stars 32 forks source link

evaluate problem #51

Closed NicholasEinstein closed 6 months ago

NicholasEinstein commented 8 months ago

作者你好，我运行你的evaluate，使用的是你提供的book test的数据，我发现会出现显存爆掉的问题 1709459902384

当将batchsize调为1以后 1709459937027

才能够正常运行，即使如此显存占用也很大，我的gpu是24gb的 1709459985248

请问这是正常的吗

SAI990323 commented 8 months ago

看起来是正常的, batchsize可以适当在加大一点，根据您的显存占用进行调整，如果我没记错的话24g的情况下batch可以开到4或8.