Question about 'Joint training of the entire pipeline and Refinement-Net'

[Climmy]

Hi, I encounter a problem when I run the command in ''Joint training of the entire pipeline and Refinement-Net'

I directly load the pre-train model trained on the synthetic data. The loss doesn't diminish during the following training and the model generates blank images after several iterations.

Some points may be helpful to figure out the cause of the problem:

1. There seems to be an inconsistency between the code and the article. According to the article, Ltv and Lp are used in the fine-tune stage.
   
   
   However, in 'finetune_real_dataset.py', I couldn't find Lp or Ltv:
   

2. In 'finetune_real_dataset.py', all weights of the four models are updated, should some of them be fixed?

Thanks a lot!