訓練參考資料¶
除了許多模型、資料集和影像轉換之外,Torchvision 還提供訓練參考腳本。 這些是我們用來訓練模型的腳本,然後可以使用預訓練權重。
這些腳本不是核心套件的一部分,而是在 GitHub 上提供。 我們目前提供分類、偵測、分割、相似性學習和視訊分類的參考資料。
雖然這些腳本在很大程度上是穩定的,但它們不提供向後相容性保證。
一般來說,這些腳本依賴於最新的(尚未發布)pytorch 版本或最新的 torchvision 版本。 這意味著要使用它們,您可能需要安裝最新的 pytorch 和 torchvision 版本,例如:
conda install pytorch torchvision -c pytorch-nightly
如果您需要依賴舊版本的穩定 pytorch 或 torchvision,例如 torchvision 0.10,那麼使用 GitHub 上相應版本的腳本更安全,即 https://github.com/pytorch/vision/tree/v0.10.0/references。