layer 별 learning rate 할당 방법
import torch import torch.nn as nn import torchvision.models as models class CNN(nn.Module): def __init__(self, ): super(CNN, self).__init__() self.backbone = models.resnet18(pretrained=True) for p in self.backbone.parameters(): p.requires_grad = True self.fc = nn.Linear(512, 2) def forward(self, x): x = self.backbone.conv1(x) x = self.backbone.bn1(x) x = self.backbone.relu(x) x = self.backbone...
Deep Learning/PyTorch
2020. 9. 23. 01:10
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- ㅂ
- 백준 11053
- 순열
- 백트래킹
- LCA
- 백준 1766
- MOT
- C++ Deploy
- 가장 긴 증가하는 부분 수열
- 단축키
- 위상 정렬 알고리즘
- PyCharm
- cosine
- 파이참
- 인공지능을 위한 선형대수
- FairMOT
- 자료구조
- 이분탐색
- 백준 11437
- 문제집
- Lowest Common Ancestor
- 조합
- 백준
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | |||
5 | 6 | 7 | 8 | 9 | 10 | 11 |
12 | 13 | 14 | 15 | 16 | 17 | 18 |
19 | 20 | 21 | 22 | 23 | 24 | 25 |
26 | 27 | 28 | 29 | 30 | 31 |
글 보관함