soyeonland

1. model 생성 import torch from torch import nn import torch.nn.utils.prune as prune import torch.nn.functional as F device = torch.device('cuda' if torch.cuda.is_available() else "cpu") class LeNet(nn.Module): def __init__(self): super(LeNet, self).__init__() self.conv1 = nn.Conv2d(1, 6, kernel_size=3) self.conv2 = nn.Conv2d(6, 16, kernel_size=3) self.fc1 = nn.Linear(16 * 5 *5, 120) self.fc2 = nn..

The purpose of the document is study class PriorBox(object): """ Compute priorbox coordinates in center-offset form for each source feature map """ def __init__(self, cfg): super(PriorBox, self).__init__() self.image_size = cfg['min_dim'] #300 self.num_priors = len(cfg['aspect_ratios']) #6 self.variance = cfg['variance'] or [0.1] self.feature_maps = cfg['feature_maps'] #[38, 19, 10, 5, 3, 1] sel..