Fix RoIFeatureTransform Methods with ArgMax
Summary: For RoIFeatureTransform Methods with ArgMax (RoIPoolF/RoIPoolLateQuantization), the current pipeline will report ``` TypeError: I/O blob #0 of unsupported type: (BlobReference("gpu_0/roi_features"), BlobReference("gpu_0/_argmax_roi_features")) of type <type 'tuple'> ``` as in the example training: https://our.intern.facebook.com/intern/fblearner/details/51149202/ Reviewed By: viswanathgs Differential Revision: D7907518 fbshipit-source-id: 38caa51c955f5a577d11c04688b9061cb6dd3c22
Showing
Please
register
or
sign in
to comment