unicorn who dev
Add on-device learning and optimizer resume: fireviewer_rfdetr_medium_v110
22e5e0f verified
Raw History Blame Contribute Delete
3.14 kB
{
"schema": 2,
"key": "fireviewer_rfdetr_medium_v110_learning",
"source_model": "fireviewer_rfdetr_medium_v110",
"task": "object_detection",
"upstream": "fireviewer/fire-smoke-rfdetr-medium-v110-strict-v1",
"upstream_revision": "901382d9c8eef2af8b4c13640413c03dbfdbeec2",
"backbone_frozen": true,
"training_scope": "candidate_classification_and_box_adaptation_only",
"report": {
"key": "fireviewer_rfdetr_medium_v110",
"validation_passed": true,
"training_scope": "candidate_classification_and_box_adaptation_only",
"backbone_frozen": true,
"limitation": "Does not create proposals absent from frozen detector",
"initial_detector_parity": true,
"parameters_changed": true,
"initial_loss": 0.8691307902336121,
"final_loss": 0.19148443639278412,
"mixed_image_shapes": [
[
224,
224
],
[
224,
288
],
[
288,
224
]
],
"restart_logits_exact": true,
"optimizer_resume_exact": true,
"variable_image_preprocessing": {
"mode": "bilinear_stretch_inside_graph",
"native_backbone_shape": [
1,
3,
576,
576
]
},
"android_device_tested": false,
"accuracy_evaluated": false,
"sha256": "dba3ade58743b422d492448605e1cb7b709de66323dd0fbdbd930f12af7bad20",
"bytes": 124411016,
"signatures": {
"infer": {
"inputs": [
"x"
],
"outputs": [
"boxes",
"detections",
"logits"
]
},
"restore": {
"inputs": [
"checkpoint_path"
],
"outputs": [
"step"
]
},
"save": {
"inputs": [
"checkpoint_path"
],
"outputs": [
"step"
]
},
"train": {
"inputs": [
"learning_rate",
"x",
"y"
],
"outputs": [
"loss",
"step"
]
},
"training_state": {
"inputs": [],
"outputs": [
"head_kernel",
"step"
]
}
},
"runtime_requirement": "LiteRT builtin CPU; disable default XNNPACK delegates; Select TF Ops required for checkpoint Save/Restore",
"android_runtime_dependencies": [
"org.tensorflow:tensorflow-lite:2.16.1",
"org.tensorflow:tensorflow-lite-select-tf-ops:2.16.1"
],
"frozen_backbone_sha256": "badcf1f0edbef2e6f319370ac7d5a2231d4332236656087f94b86376e73efcf9",
"head_initialization": "source_model_or_identity_adapter",
"conversion_source_sha256": {
"merge_training_graph.py": "93def8b746dd763c7ef703e29a5db3bab754716c6358bdda5acb64531b4c50ce",
"build_trainable_detector.py": "4e7794e81cd426850a1f84ee5c6b5af92316a2b62db10461e9f91bc896c7bb30"
},
"source_parity_report": "source_parity.json"
},
"optimizer": "momentum SGD, momentum=0.9",
"checkpoint_contains": [
"trainable parameters",
"momentum buffers",
"global step"
],
"requires_select_tf_ops": true,
"android_configuration": "android_model_config.json"
}