The error is the same you posted (Blob is not a CPU Tensor), it refers to the blob just before the output blob and it is a Dropout module in my network. If I remove the Dropout module from the original network architecture, the error is gone and everything executes well. I’ve seen reports of the same behavior before: https://github.com/facebookarchive/caffe2/issues/2165 8.
Thanks.