code for the fitting procedure

[ThibaultGROUEIX]

Hi,
Thanks for a great job!
Is there any plan to release to code for the fitting procedure? (or I may have missed it?)
Best regards,
-Thibault

[jhugestar]

Hi Thibault,

Yes, we plan to do it hopefully after Thanksgiving break!

[lawy623]

hi

Have you released the core code for fitting?

[dumyy]

emmmm,the same issue.

[littlejiumi]

hi

Have you released the core code for fitting?

Thanks for replying.

[jhugestar]

Sorry for the super delay!

You can find new instruction:

https://github.com/facebookresearch/eft/blob/eft_fit/README_run_eft.md

[littlejiumi]

Thanks for your instruction!
I have the following error:

Can you tell me how to fix it?

[littlejiumi]

Thanks for your instruction!
I have the following error:

Can you tell me how to fix it?

Fix it by using one single gpu.

[uyoung-jeong]

Instead of using single gpu, I manually added 'module.' prefix on reloadModel function:

eft/eft/cores/base_trainer.py

Line 209 in c8400a6

def reloadModel(self):

new_dict = OrderedDict()
for k,v in self.model_backup.items():
    new_k = k
    if not 'module.' in k:
        new_k = 'module.' + k
    new_dict[new_k]=v
self.model.load_state_dict(new_dict)