bugfix: fit_XC_XY was not working with decoupled/pretrain_finetune because input vars to train_fn were wrong
bugfix: fit_XC_XY was not working with decoupled/pretrain_finetune because input vars to train_fn were wrong

tu-rbo/concarne