I think that the parameter called msp_weight in the docs and example configs is called msp.alpha in the actual code. I don't see msp_weight used anywhere in the actual code. So I'm recommending this doc change. But if this is not right please let me know!
But in any case, without specifying msp.alpha, models of type cond-ae-msp cannot be fit, so the default config file doesn't work out of the box. That's what led me to look into this.
I think that the parameter called
msp_weight
in the docs and example configs is calledmsp.alpha
in the actual code. I don't seemsp_weight
used anywhere in the actual code. So I'm recommending this doc change. But if this is not right please let me know!But in any case, without specifying msp.alpha, models of type cond-ae-msp cannot be fit, so the default config file doesn't work out of the box. That's what led me to look into this.