Currently, checkpoint function allows only Tensor/Variable output. But some architectures might have more than one output, such as the encoder of a standard FCNs, which might have several outputs from the mainstream and skip connection, or architectures that have both text and label outputs.
Now, my solution is to concatenate the tensors in the function and unroll them outside checkpoint, which is not so convenient.
So are there any neat way to do that? Or would it be list supports in the future?
Thank you so much, really appreciate your work !