이번 예제는 VGGNet 중에서 가장 간단한 VGG11을 파이토치로 구현합니다. VGG16이나 VGG19를 구현해 보고 싶다면, VGG11에서 더 깊게 쌓아 올리면 됩니다.
가장 처음으로 할 일은 필요한 라이브러리를 호출하는 것입니다.
코드 6-39 필요한 라이브러리 호출
import copy ------ ①
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torch.utils.data as data
import torchvision
import torchvision.transforms as transforms
import torchvision.datasets as Datasets
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
① 객체 복사를 위해 사용합니다. 객체 복사는 크게 얕은 복사(shallow copy)와 깊은 복사(deep copy)로 나뉩니다. 예제를 통해 살펴보겠습니다.