Can you share engine_for_Finetuning.py? I found it is missing when fine-tuning downstream tasks.

[Jiaqi2Wx]

There seems to be some problems in the code, and the model cannot be loaded in run_class_finetuning.py.

[ayumiymk]

I am sorry that I only checked the pre-training part before.
Now, I have update the files and you can go ahead.

[EternityWM]

I am sorry that I only checked the pre-training part before. Now, I have update the files and you can go ahead.

I'm sorry I still can't find this file.Can you give me some specific guidance?
Thank you !

[Jiaqi2Wx]

https://github.com/ayumiymk/DiG/tree/master
it still has some 'import' bugs in run_class_finetuning.py. Can u fix them.

[Jiaqi2Wx]

I am sorry that I only checked the pre-training part before. Now, I have update the files and you can go ahead.

I'm sorry I still can't find this file.Can you give me some specific guidance? Thank you !

not in main, in master

[EternityWM]

I am sorry that I only checked the pre-training part before. Now, I have update the files and you can go ahead.

I'm sorry I still can't find this file.Can you give me some specific guidance? Thank you !

not in main, in master

AaH,Thanks for your reply

[ayumiymk]

https://github.com/ayumiymk/DiG/tree/master it still has some 'import' bugs in run_class_finetuning.py. Can u fix them.

Ok, I will check it as soon as possible.

[ayumiymk]

I have carefully double-checked the code of fine-tune and evaluation. Now you can use the master branch.
Latter, I will merge the master branch to the main branch.