Wuyxin / DISC

Official code of "Discover and Cure: Concept-aware Mitigation of Spurious Correlation" (ICML 2023)

Home Page:https://arxiv.org/abs/2305.00650

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training issue - name 'is_training' is not defined

AmitRozner opened this issue · comments

When trying to run the training I face the following issue:
disc/disc/utils/loss.py", line 198, in get_stats
group_str = self.group_str(idx, is_training) if 'Meta' in self.args.dataset else self.group_str(idx)
NameError: name 'is_training' is not defined

Can you assist? This is indeed not defined so I am unsure how it worked for you.
Thank you!

Hi,

You are right, this is a bug which we didn't notice when cleaning our codebase. The reason is that we were testing on CUB dataset and the bug only triggers on MetaShift dataset. Apologies for the confusion. I believe it's fixed rn.

Thanks a lot for letting me know!

Close this issue after the fix, pls let me know if there are more Qs!