pytorch 调用模型

利用pytorch调用预训练好的模型到GPU上
modeldata = torch.load(‘resnet-34-kinetics.pth’)

利用pytorch调用预训练好的模型到CPU上
modeldata = torch.load(‘resnet-34-kinetics.pth’,map_location=lambda storage, loc: storage)
返回值modeldata是一个Dict形变量

import torch
#torch.load with map_location='cpu'
modeldata = torch.load('/Users/yunshanjinwen/Downloads/video-classification-3d-cnn-pytorch-master/resnet-34-kinetics.pth',map_location=lambda storage, loc: storage)
print(modeldata)

以下为部分结果截图：

{'epoch': 251, 'arch': 'resnet-34', 'state_dict': OrderedDict([('module.conv1.weight', tensor([[[[[-1.6517e-04, -1.9815e-03,  6.5790e-05,  ..., -1.7464e-03,
             7.1315e-04,  2.7513e-03],
           [ 4.5178e-04, -2.0219e-03,  2.4164e-03,  ..., -2.5019e-03,
            -1.2443e-03,  5.2268e-03],
           [-2.1355e-05, -4.8858e-03,  1.7820e-03,  ..., -2.1954e-03,
            -6.8539e-03,  3.0589e-03],
           ...,
           [ 1.9253e-03, -8.9324e-03, -5.0934e-03,  ...,  1.0542e-02,
            -1.5762e-02, -1.0906e-02],
           [ 5.8645e-03, -5.7808e-03, -6.5108e-03,  ...,  1.3623e-02,
            -1.0710e-02, -1.0448e-02],
           [ 5.6550e-03, -2.1331e-03, -5.4692e-03,  ...,  9.5111e-03,
            -9.2105e-03, -1.0438e-02]],

          [[-1.3501e-03, -1.9448e-03,  5.9943e-04,  ..., -9.1701e-03,
            -5.9438e-03,  1.9499e-03],
           [ 6.4719e-04,  1.1618e-03,  9.6282e-03,  ..., -1.2365e-02,
            -1.3114e-02,  1.2741e-03],
           [-2.8409e-03, -2.0466e-03,  1.5195e-02,  ..., -8.6988e-03,
            -2.0623e-02, -2.6702e-03],
           ...,
           [-6.5846e-03, -1.2465e-02,  9.3114e-03,  ...,  1.8109e-02,
            -3.0516e-02, -2.4903e-02],
           [-2.7478e-03, -1.2591e-02,  7.6693e-04,  ...,  2.6830e-02,
            -1.7754e-02, -2.1244e-02],
           [-1.1187e-03, -7.1811e-03,  8.5726e-04,  ...,  2.4925e-02,
            -9.5690e-03, -1.7701e-02]],

          [[-3.7975e-04, -1.7536e-03,  7.3226e-04,  ..., -8.5807e-04,
            -6.5082e-03, -1.6441e-03],
           [ 1.7633e-04, -1.4810e-03,  2.2259e-03,  ..., -2.3171e-03,
            -6.1884e-03,  1.8897e-03],
           [-1.8017e-03, -4.1376e-03,  3.9245e-03,  ..., -9.8589e-04,
            -7.4184e-03,  2.7875e-03],
           ...,
           [-4.2223e-03, -1.0657e-02, -3.5849e-03,  ...,  6.0524e-03,
            -1.1682e-02, -2.2864e-03],
           [ 1.4787e-03, -6.3200e-03, -4.7488e-03,  ...,  9.9818e-03,
            -6.3154e-03, -2.6875e-03],
           [ 2.6545e-03, -3.0796e-03, -3.8646e-03,  ...,  7.4478e-03,
            -3.7495e-03, -2.2428e-03]],

          ...,

          [[-2.1119e-03, -4.5376e-03, -3.5092e-03,  ...,  6.7190e-03,
            -3.1279e-03,  8.5857e-04],
           [ 1.0150e-03,  3.3156e-04, -5.7779e-03,  ...,  1.1072e-02,
             5.0206e-03,  1.4566e-03],
           [ 2.4621e-03,  4.6832e-03, -5.9863e-03,  ...,  1.2540e-02,
             1.3816e-02,  1.5569e-03],
           ...,
           [ 1.4451e-03,  1.3931e-02, -4.5525e-03,  ..., -7.6014e-03,
             2.6069e-02,  1.6237e-02],
           [ 3.3758e-03,  1.4173e-02,  3.9455e-03,  ..., -1.5280e-02,
             1.8754e-02,  1.8436e-02],
           [ 1.9167e-03,  4.4112e-03, -2.0688e-03,  ..., -1.7529e-02,
             1.0156e-02,  1.9978e-02]],

          [[-9.3798e-04, -2.9649e-03, -1.0596e-03,  ...,  5.9191e-03,
            -4.1590e-03, -4.0802e-04],
           [ 1.0737e-03,  1.8740e-03, -1.1678e-03,  ...,  7.6876e-03,
             3.680