pytorch之hub模組

星空28發表於2024-10-05

hub模組中有前人已經訓練好的模型引數,可以根據情況呼叫

# =========================hub模組:呼叫============================ #
torch.hub.list('pytorch/vision:v0.10.0')
# 使用前人已有的模型庫
# https://github.com/pytorch/hub/blob/master/pytorch_vision_resnet.md

相關文章