Zhangjinso / PISE

123 stars 28 forks source link

Respective / end-to-end training #37

Open Archjbald opened 2 years ago

Archjbald commented 2 years ago

Hello,

First of all, thank you for your work. I read in your paper that you first train the parsing generator and image generator respectively, then perform an end-to-end training. However, I was not able to locate in your code the parts that handle the switch of training strategy. The reason for this question is that I would like to train only the parsing generator first and would not like to change everything in the code if the option is already there.

Thank you for your help

Zhangjinso commented 2 years ago

You can use current setting to train your model.

Archjbald commented 2 years ago

Ok, but can you confirm that by using the current code / settings as so, the training is performed end to end all along?