fine-tuning the model in the third step of training,

[gwx541]

When fine-tuning the model in the third step of training, should the config of the main.py file be changed to the final_checkpoint.pth file in Pre-training and the train_eps_10.0_s.yaml file in PRIVIMAGE+D? After I changed it, opacus reported an error.

[SunnierLee]

Hi! You also need to change the "data.path" and "data.fid_stats" into yours in train_eps_10.0_s.yaml. Could you show me the error?

[gwx541]

I had changed data.pth、data.fid_satas and model.ckpt The error is following

[gwx541]

Hi! You also need to change the "data.path" and "data.fid_stats" into yours in train_eps_10.0_s.yaml. Could you show me the error?
I had show the error.

[SunnierLee]

1. Please run the following command to reinstall opacus:

cd xxx/DP-ImaGen
pip uninstall opacus
pip install -e src/opacus

2. Please replace your DP-ImaGen/src/PRIVIMAGE+D/runners/train_dpdm_base.py and DP-ImaGen/src/PRIVIMAGE+D/model/layers.py with the latest files in the repo.

Besides, we fixed a bug in query_semantics.py yesterday. If you ran query_semantics.py before, you may need to update query_semantics.py, re-run it ,and then pre-train your models! You can check if the synthetic images generated by the pre-trained models are normal.