訓練參考¶
除了提供眾多模型、資料集和影像變換外,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。