no attribute 'pretrain_ema'

jorghyq2016 / FedHSSL

The implementation of FedHSSL algorithm published in the paper "A Hybrid Self-Supervised Learning Framework for Vertical Federated Learning".

13 stars 1 forks source link

no attribute 'pretrain_ema' #1

Closed wg-li closed 1 year ago

wg-li commented 1 year ago

Hello, Nice work. Thanks for the code, but I got this error:

File "/data/nfs/user/liwg/vfl/fedhssl/FedHSSL/pretrain_template.py", line 142, in update_local_top_model if self.args.pretrain_ema > 0.0: AttributeError: 'Namespace' object has no attribute 'pretrain_ema'

I didn't find "pretrain_ema" defined in the code. Could you please check it?

jorghyq2016 commented 1 year ago

Hi,

sorry for the inconvience. The "pretrain_ema" is an old parameter. Now it should be fine. Let me know if you have other problems. Best.

yankang18 commented 1 year ago

issue addressed

wg-li commented 1 year ago

thanks!