Parameter max_decode_args

[vthost]

I am using a different dataset with larger molecules, however this breaks the code (e.g. the tree walker has too large data structures in batch_make_att_masks). I see cmdargs.max_decode_steps and would assume that molecules of larger size are ignored but this doesn't seem to be the case. Is there maybe another parameter I have to set? Thank you already!