반응형
resnet18
-
[Pytorch] transfer_Learning 전이학습딥러닝(deep Learning) 2021. 5. 17. 02:22
0. 들어가기 전에 - 파이토치의 전이학습 을 통하여 모델은 Resnet18 로 벌과 개미를 구분하는 모델을 만들어보겠습니다. 1. 모듈 임포트 In [103]: from __future__ import print_function, division import torch import torch.nn as nn import torch.optim as optim from torch.optim import lr_scheduler import numpy as np import torchvision from torchvision import datasets, models, transforms import matplotlib.pyplot as plt import time import os import copy pl..