现在Trainer每次epoch完后都会保存模型checkpoint 其中STMLP会自动教师模型到pre-train 根据教师模型的存在情况启动/预训练or蒸馏模式 |
||
|---|---|---|
| .. | ||
| cdeTrainer | ||
| DCRNN_Trainer.py | ||
| PDG2SEQ_Trainer.py | ||
| STMLP_Trainer.py | ||
| Trainer.py | ||
| trainer_selector.py | ||
现在Trainer每次epoch完后都会保存模型checkpoint 其中STMLP会自动教师模型到pre-train 根据教师模型的存在情况启动/预训练or蒸馏模式 |
||
|---|---|---|
| .. | ||
| cdeTrainer | ||
| DCRNN_Trainer.py | ||
| PDG2SEQ_Trainer.py | ||
| STMLP_Trainer.py | ||
| Trainer.py | ||
| trainer_selector.py | ||