Skip to content
Snippets Groups Projects
Commit 1e8b3320 authored by Hyunjun Go's avatar Hyunjun Go
Browse files

Update README.md

parent 8a2124e2
No related branches found
No related tags found
No related merge requests found
......@@ -22,3 +22,31 @@ Deep learning에 사용되는 프레임워크인 PyTorch를 통해
#### tutorial code
[ipynb 파일 참조](timm_tutorial.ipynb)
#### timm construct model
* GPU 사용의 경우 ```.cuda()``` 혹은 ```.to(divice)``` 필요
```python
model = timm.create_model(
'모델명', pretrained=True, num_classes=num_classes
)
```
#### timm 구현된 model list 출력
```python
import timm
from pprint import pprint
model_names = timm.list_models(pretrained=True)
pprint(model_names)
```
#### timm construct optimizer
* GPU 사용의 경우 ```.cuda()``` 혹은 ```.to(divice)``` 필요
```python
optimizer = timm.optim.create_optimizer_v2(model, '옵티마이저 명', lr=learning_rate)
```
default = SGD(stochastic gradient descent) optimizer
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment