allenai / allenai/rslearn

[Enhancement] Benchmark data for generating embeddings

未关闭
#579 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
93
派生
15
平均合并
20 小时 56 分钟
30 天内合并 PR
4

描述

Proposal is to augment the docs with benchmark info for embeddings generation, simultaneously documenting performance on systems such as apple Silicon (mps). Could be similar to https://omnicloudmask.readthedocs.io/en/latest/performance.html

Noting MPS use requires some additional config since rslearn passes trainer options directly to Lightning and sets a default DDP strategy, but Lightning DDP is not supported on MPS.

```json
trainer:
accelerator: mps
devices: 1
strategy: auto
```

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。