In plain PyTorch you could simply return crys_fea with out in your forward method.
I’m not sure, if this will work in skorch or if you are limited to a single return value.
Could you try that and see if it’s working?