论文

OlmoEarth v1.2:更高效的 OlmoEarth 模型系列

OlmoEarth v1.2: A more efficient family of OlmoEarth models

模型推理推理加速

摘要

我们对 OlmoEarth 系列进行了一系列改进。这些改进使我们能够削减训练期间的计算成本(训练基本模型所需的 GPU 时间减少 3.0 美元×$)和推理(Sentinel-2 任务的 MAC 时间减少 2.9 美元×$),同时保持模型的整体性能。所有训练代码均可在 github.com/allenai/olmoearth_pretrain 上获取。