Notice
Recent Posts
Recent Comments
Link
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
Tags
- label
- pytorch
- pandas
- C#
- 채보
- Python
- nvidia-smi
- windows forms
- paramiko
- 핑거스타일
- SSH
- mysql
- Docker
- error
- pip
- 프로그래머스
- Selenium
- 오류
- C
- 컨테이너
- OpenCV
- VS Code
- ubuntu
- YOLO
- JSON
- 기타 연주
- Linux
- C++
- Numpy
- Visual Studio
Archives
- Today
- Total
목록model (1)
기계는 거짓말하지 않는다
PyTorch model save and load
import torchmodel = # 모델 생성# 모델 저장save_name = "model_final"torch.save(model.state_dict(), f"Directory/{save_name}.pth")# 모델 불러오기load_name = "model_final"model.load_state_dict(torch.load(f"Directory/{load_name}.pth")) # 경로, 저장 모델 이름# 설정 값 세부 저장checkpoint = { 'model': model.state_dict(), 'optimizer': optimizer.state_dict(), 'lr_scheduler': lr_scheduler.state_dict(), 'epoch': epoch}torch.save(c..
AI
2021. 8. 8. 16:32