This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Save teacher weight #64

Open

ShenZheng2000 opened this issue May 26, 2023 · 1 comment

ShenZheng2000 commented May 26, 2023

The current code only saves the student' weight. How to save the teacher weight during training?

Contributor

yujheli commented Jun 5, 2023

The code should save both. It saves one checkpoint containing two models.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.