FeatureStyleEncoder
FeatureStyleEncoder copied to clipboard
Error in the training code
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.