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 摘要。