pytorch之hub模块

hub模块中有前人已经训练好的模型参数,可以根据情况调用

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

posted @ 2024-10-05 14:49  星空28  阅读(10)  评论(0编辑  收藏  举报