訓練參考¶
除了許多模型、資料集和影像轉換之外,Torchvision 還提供了訓練參考指令碼。這些是我們用於訓練 模型 的指令碼,這些模型隨後可提供預訓練權重。
這些指令碼不是核心包的一部分,而是 在 GitHub 上 提供。我們目前為 分類、檢測、分割、相似性學習 和 影片分類 提供參考。
雖然這些指令碼在很大程度上是穩定的,但它們不提供向後相容性保證。
總的來說,這些指令碼依賴於最新的(尚未釋出的)pytorch 版本或最新的 torchvision 版本。這意味著,要使用它們,您可能需要按照 官方說明 安裝最新的 pytorch 和 torchvision 版本。
如果您需要依賴舊的穩定版 pytorch 或 torchvision,例如 torchvision 0.10,那麼最好使用 GitHub 上該對應版本的指令碼,即 https://github.com/pytorch/vision/tree/v0.10.0/references。