zoukankan      html  css  js  c++  java
  • vgg等模型下载很慢

    vgg等模型由于权限或网速原因无法在云服务器正常下载

    可以先下载好放到本地(有时候在云服务器上没有权限下载)

    import torch
    import torch.nn as nn
    import torch.nn.functional as F
    from torchvision import models
    
    
    #vgg_pretrained_features = models.vgg19(pretrained=True).features #直接下载
    #无法直接下载
    vgg_pretrained = models.vgg19(pretrained=False)
    vgg_pretrained.load_state_dict(torch.load('/.../vgg19-dcbb9e9d.pth'))#load内为本地路径
    vgg_pretrained_features = vgg_pretrained.features
  • 相关阅读:
    2020/5/8
    2020/5/8
    2020/5/6
    2020/4/30
    2020/4/29
    2020/4/28
    2020/4/27
    KMP算法详解
    博客搬家声明
    洛谷P2831 NOIP2016 愤怒的小鸟
  • 原文地址:https://www.cnblogs.com/tingtin/p/14609374.html
Copyright © 2011-2022 走看看