```py
# 1. Directly Load a Pre-trained Model
# https://github.com/pytorch/vision/tree/master/torchvision/models
import torchvision.models as models
resnet50 = models.resnet50(pretrained=True)
# or
model = models.resnet50(pretrained=False)
# Maybe you want to modify the last fc layer?
resnet.fc = nn.Linear(2048, 2)
# 2. Load part of parameters of a pretrained model as init for self-defined similar-architecture model.
# resnet50 is a pretrain model
# self_defined indicates model you just define.
resnet50 = models.resnet50(pretrained=True)
self_defined = ...
pretrained_dict = resnet50.state_dict()
model_dict = self_defined.state_dict()
pretrained_dict = {k: v for k, v in pretrained_dict.items() if k in model_dict}
# update & load
model_dict.update(pretrained_dict)
model.load_state_dict(model_dict)
# 3. Save & Load routines.
# routine
# (recommended) saves and loads only the model parameters
torch.save(model.state_dict(), PATH)
model.load_state_dict(torch.load(PATH))
# routine 2
# saves and loads the entire model
torch.save(model, PATH)
model = torch.load(PATH)
```
- Welcome
- fzf使用
- git回滾
- 安裝gollum使用github作為wiki端
- 使用github的issue
- linux的腳本
- git沖突解決
- sphinx生成python文檔
- 使用aria2作為下載客戶端
- tmux使用
- anaconda的虛擬環境構建
- git commit message
- tensorflow指定GPU,限制顯存
- pycharm professional
- 好用的命令行工具
- lrzsz使用
- 使用iTerm2的profile免密登陸遠程主機
- quicklook 插件
- docker基本使用
- jupyter notebook轉html和pdf
- jupyter-notebook幾個有用命令
- mongodb安裝
- pytorch載入與訓練模型
- git覆蓋分支
- Jetson TX2資源
- GitHub上fork的repo和上游分支保持一致的方法
- vscode調試前端代碼
- github hosts
- git添加遠程倉庫
- 其它語言好用的庫/包
- git強制拉取和強制提交
- chrome擴展發布
- mac的CPP lib
- vscode遠程