Batch processing for a custom python layer

10 views
Skip to first unread message

Alex Ter-Sarkisov

unread,
Nov 26, 2017, 6:53:09 PM11/26/17
to Caffe Users
I've written my Python layer that transforms the ground truth labels. Looking at other custom python layers (e.g. rpn) I didn't quite understand if I need to add the functionality for processing batches size >1 and outputting the per-image loss for the batch as a vector. Could someone explain perhaps? 
Reply all
Reply to author
Forward
0 new messages