yeungchenwa
Hello, I am very interested in your research after reading your paper. But I have a confusion about the second stage of training.
Hello, I am very interested in your research after reading your paper. But I have a confusion about the second stage of training.
May I ask how to load the model from the first stage during the second stage of training? Is it the total model. pth from the first stage? But it seems that it cannot be loaded. If the downloaded scr_210000.pth can be loaded, what is the relationship between this and the first stage of my own training. Please help answer, thank you very much!!
hi@jingmingtao. Feel sorry that the previous codes may confuse you, and I update the codes. During phase 2, the trained parameters obtained in phase 1 will be resumed in phase 2. You can turn phase_1_ckpt_dir to your checkpoint files directory obtained in phase 1.
