def __init__(self, net, init_net, parameters, int_features) -> None:
RLPredictor.__init__(self, net, init_net, parameters, int_features)
self._output_blobs = [
"output/float_features.lengths",
"output/float_features.keys",
"output/float_features.values",
]
def __init__(self, net, init_net, parameters, int_features):
RLPredictor.__init__(self, net, init_net, parameters, int_features)
self._output_blobs.extend([
OUTPUT_SINGLE_CAT_KEYS_NAME,
OUTPUT_SINGLE_CAT_LENGTHS_NAME,
OUTPUT_SINGLE_CAT_VALS_NAME,
])
def __init__(self, net, init_net, parameters, int_features):
RLPredictor.__init__(self, net, init_net, parameters, int_features)
self._output_blobs.extend(
[
OUTPUT_SINGLE_CAT_KEYS_NAME,
OUTPUT_SINGLE_CAT_LENGTHS_NAME,
OUTPUT_SINGLE_CAT_VALS_NAME,
]
)
