This repository has been archived on 2026-03-23. You can view files and clone it, but cannot push or open issues or pull requests.
Zengwei Yao c059ef3169
Keep model_avg on cpu (#348)
* keep model_avg on cpu

* explicitly convert model_avg to cpu

* minor fix

* remove device convertion for model_avg

* modify usage of the model device in train.py

* change model.device to next(model.parameters()).device for decoding

* assert params.start_epoch>0

* assert params.start_epoch>0, params.start_epoch
2022-05-07 10:42:34 +08:00
..
2022-05-07 08:10:54 +08:00
2022-04-22 15:54:59 +08:00