debugging beam train
fixed oneofmany task -> argmax task conversion