现在Trainer每次epoch完后都会保存模型checkpoint 其中STMLP会自动教师模型到pre-train 根据教师模型的存在情况启动/预训练or蒸馏模式 |
||
|---|---|---|
| .. | ||
| Download_data.py | ||
| Trainer_old.py | ||
| initializer.py | ||
| logger.py | ||
| loss_function.py | ||
| normalization.py | ||
现在Trainer每次epoch完后都会保存模型checkpoint 其中STMLP会自动教师模型到pre-train 根据教师模型的存在情况启动/预训练or蒸馏模式 |
||
|---|---|---|
| .. | ||
| Download_data.py | ||
| Trainer_old.py | ||
| initializer.py | ||
| logger.py | ||
| loss_function.py | ||
| normalization.py | ||