FeatureStyleEncoder icon indicating copy to clipboard operation
FeatureStyleEncoder copied to clipboard

Error in the training code

Open Xu-Yao opened this issue 2 years ago • 0 comments

The training code in trainer.py line 180 (features = [None]*k + [fea] + [None]*(17-k))should be modified as:

if training_mode:
    features = None
else:
    features = [None]*k + [fea] + [None]*(17-k)

I have received an email that identifies this problem and proposes a modification. This modifications has no impact at test time but do impact the training. Without this the model is not trained on two inversions.

Xu-Yao avatar Mar 23 '23 10:03 Xu-Yao