苏凯祺的博客

def show_feature_map(feature_map): feature_map = feature_map.squeeze(0) feature_map = feature_map.cpu().numpy() feature_map_num = feature_map.shape[0] row_num = np.ceil(np.sqrt(feature_map_num)) plt.figure() for index in range(1, f.

import torchimport torch.nn as nnimport torch.optim as optimimport torchvisionimport torchvision.transforms as transformsimport argparsefrom resnet import resnet50# 定义是否使用GPUdevice = torch.d...

import torchimport torch.nn as nnimport torch.optim as optimimport torchvisionimport torchvision.transforms as transformsimport argparsefrom resnet18 import ResNet18# 定义是否使用GPUdevice = torch...

import torchimport torch.nn as nn#from .utils import load_state_dict_from_url#其中torch.nn 为其提供基础函数，model_zoo提供权重数据的下载。__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101', ...

# coding=utf-8import torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd import Variableimport torchimport torchvisionimport torchvision.transforms as transformsimport torch.optim...