chrome_models / screen_ai /screen2x_config.pbtxt
dejanseo's picture
Upload 99 files
993bb90 verified
raw
history blame
1.27 kB
# proto-file: knowledge/cerebra/sense/im2query/screenai/screen2x/lib/screen2x.proto
# proto-message: screenai.screen2x.Screen2xOptions
model_path: "knowledge/cerebra/sense/im2query/screenai/screen2x/lib/data/chrome_a11y/v011/model.tflite"
model_config {
mpnn_config {
signature: "serving_default"
senders_node_name: "input_1_senders"
receivers_node_name: "input_1_receivers"
nodes_node_name: "input_1_nodes"
edges_node_name: "input_1_edges"
globals_node_name: "input_1_globals"
n_node_node_name: "input_1_n_node"
n_edge_node_name: "input_1_n_edge"
numeric_features_node_name: "input_2_numeric"
chrome_role_node_name: "input_2_chrome_role"
chrome_role_description_node_name: "input_2_chrome_role_description"
output_node_name: "output_1"
inference_config {
num_threads: 1
class_thresholds {
label: CONTENT_NON_ESSENTIAL
prediction_threshold: 0.65
}
class_thresholds {
label: CONTENT_TEXT_HEADLINE
prediction_threshold: 0.0
}
class_thresholds {
label: CONTENT_TEXT_MAIN
prediction_threshold: 0.0
}
}
model_type: CHROME_A11Y
}
}
post_process_options {
chrome_post_process: true
}