业余 ML 研究者求助：在 A100 上跑 NBFNet 的低成本训练方案

一位业余研究者在 Google Vertex 的 A100 80GB 上训练 NBFNet 做药物重定位研究，因模型每查询需约 1 GB 激活显存，batch size 始终从 512 自动减半至 64。