def get_indices(tid, si): return set( Mask.get_mask_stats_by_idx( model.sess, mask_id, feed_dict={ X: model.trainXs[tid][si:si + batch_size_per_task], is_training: True, })["indices"])
def get_indices(tid, start_idx, batch_size, mask_ids, X, is_training): return set(Mask.get_mask_stats_by_idx(model.sess, mask_ids, feed_dict={ X: model.trainXs[tid][start_idx:start_idx + batch_size], is_training: True, })["indices"])