| { |
| "architectures": [ |
| "FalconModel" |
| ], |
| "auto_map": { |
| "AutoConfig": "configuration_falcon.FalconConfig", |
| "AutoModel": "modeling_falcon.FalconModel" |
| }, |
| "checkpoint_metadata": { |
| "checkpoint_format": "falcon-checkpoint-v1", |
| "config": { |
| "detector": { |
| "mask_threshold": 0.5, |
| "max_regions": 100, |
| "nms_threshold": 0.6, |
| "resolution": 768, |
| "score_threshold": 0.15, |
| "variant": "seg-2xlarge" |
| }, |
| "experiment": { |
| "backbone_layout": "independent", |
| "name": "falcon-x", |
| "protocol": "dataset-v1", |
| "ssa_ablation": "none", |
| "world_size": null |
| }, |
| "model": { |
| "component_order": [ |
| "detonator", |
| "explosive", |
| "battery" |
| ], |
| "image_size": 448, |
| "language_model": "lmsys/vicuna-7b-v1.5", |
| "link_order": [ |
| [ |
| "battery", |
| "detonator" |
| ], |
| [ |
| "battery", |
| "explosive" |
| ], |
| [ |
| "detonator", |
| "explosive" |
| ] |
| ], |
| "lora_alpha": 32, |
| "lora_dropout": 0.05, |
| "lora_rank": 16, |
| "max_regions": 100, |
| "patch_size": 14, |
| "region_dim": 1024, |
| "roi_size": 4, |
| "vision_model": "facebook/dinov2-large" |
| }, |
| "stages": { |
| "stage1": { |
| "batch_size": 1, |
| "encoder_learning_rate": 0.00015, |
| "epochs": 12, |
| "expanded_scales": false, |
| "gradient_accumulation": 16, |
| "learning_rate": 0.0001, |
| "lr_scheduler": "cosine", |
| "multi_scale": false, |
| "precision": "backend_mixed", |
| "tf32": true, |
| "warmup_epochs": 0.0, |
| "weight_decay": 0.0001 |
| }, |
| "stage2": { |
| "batch_size": 4, |
| "epoch_size": null, |
| "epochs": 1, |
| "gradient_accumulation": 4, |
| "learning_rate": 0.0001, |
| "link_loss_weight": 0.5, |
| "max_text_tokens": 256, |
| "precision": "bf16", |
| "presence_loss_weight": 0.5, |
| "risk_loss_weight": 1.0, |
| "sampling": "all", |
| "seed": 42, |
| "text_overflow_policy": "error", |
| "tf32": true, |
| "warmup_ratio": 0.03, |
| "weight_decay": 0.0 |
| }, |
| "stage3": { |
| "batch_size": 1, |
| "epoch_size": null, |
| "epochs": 1, |
| "gradient_accumulation": 16, |
| "learning_rate": 1e-05, |
| "link_loss_weight": 0.5, |
| "max_text_tokens": 256, |
| "precision": "bf16", |
| "presence_loss_weight": 0.5, |
| "risk_loss_weight": 1.0, |
| "sampling": "all", |
| "seed": 42, |
| "text_overflow_policy": "error", |
| "tf32": true, |
| "warmup_ratio": 0.03, |
| "weight_decay": 0.0 |
| } |
| }, |
| "training": { |
| "batch_size": 1, |
| "epoch_size": null, |
| "epochs": 1, |
| "gradient_accumulation": 16, |
| "learning_rate": 0.0001, |
| "link_loss_weight": 0.5, |
| "max_text_tokens": 256, |
| "precision": "bf16", |
| "presence_loss_weight": 0.5, |
| "risk_loss_weight": 1.0, |
| "sampling": "all", |
| "seed": 42, |
| "text_overflow_policy": "error", |
| "tf32": true, |
| "warmup_ratio": 0.03, |
| "weight_decay": 0.0 |
| } |
| }, |
| "dataset_categories": [ |
| { |
| "id": 1, |
| "name": "detonator" |
| }, |
| { |
| "id": 2, |
| "name": "explosive" |
| }, |
| { |
| "id": 3, |
| "name": "battery" |
| } |
| ], |
| "diagnostic_training_reasons": [], |
| "grounding_coverage": { |
| "full_match": 177693, |
| "ground_truth_instances": 238792, |
| "language_supervision_masked": 1405, |
| "matched_instances": 237385, |
| "partial_match": 496, |
| "true_negative": 6, |
| "unmatched_positive": 909 |
| }, |
| "grounding_policy": { |
| "iou_threshold": 0.5, |
| "unmatched": "mask_language_loss", |
| "version": "falcon-grounding-match/cardinality-iou-v1" |
| }, |
| "observed_supervision_coverage": { |
| "images": 34826, |
| "links": [ |
| 0, |
| 0, |
| 0 |
| ], |
| "presence": [ |
| 34826, |
| 34826, |
| 34826 |
| ], |
| "risk": 0 |
| }, |
| "official_result_eligible": true, |
| "precision": "bf16", |
| "require_observed_supervision": true, |
| "safety_capabilities": { |
| "links": [ |
| false, |
| false, |
| false |
| ], |
| "presence": [ |
| true, |
| true, |
| true |
| ], |
| "risk": false |
| }, |
| "ssa_ablation": "none", |
| "stage": 3, |
| "structured_loss_recipe": { |
| "links": "sigmoid_l1", |
| "presence": "bce_with_logits_fp32", |
| "reduction": "mean_over_finite_enabled_targets", |
| "risk": "sigmoid_l1" |
| }, |
| "supervision_capabilities": { |
| "links": [ |
| false, |
| false, |
| false |
| ], |
| "presence": [ |
| true, |
| true, |
| true |
| ], |
| "risk": false |
| }, |
| "training_complete": true |
| }, |
| "core_config": { |
| "image_size": 448, |
| "language_dim": 4096, |
| "link_loss_weight": 0.5, |
| "mask_probability_threshold": 0.5, |
| "max_regions": 100, |
| "max_text_tokens": 256, |
| "patch_size": 14, |
| "presence_loss_weight": 0.5, |
| "region_dim": 1024, |
| "risk_loss_weight": 1.0, |
| "roi_size": 4, |
| "safety_capabilities": { |
| "links": [ |
| false, |
| false, |
| false |
| ], |
| "presence": [ |
| true, |
| true, |
| true |
| ], |
| "risk": false |
| }, |
| "text_overflow_policy": "error", |
| "vision_dim": 1024, |
| "vision_prefix_tokens": 1 |
| }, |
| "detector_config": { |
| "architecture": { |
| "amp": true, |
| "bbox_reparam": true, |
| "ca_nheads": 16, |
| "cls_loss_coef": 1.0, |
| "dec_layers": 6, |
| "dec_n_points": 2, |
| "encoder": "dinov2_windowed_small", |
| "gradient_checkpointing": false, |
| "group_detr": 13, |
| "hidden_dim": 256, |
| "ia_bce_loss": true, |
| "layer_norm": true, |
| "license": "Apache-2.0", |
| "lite_refpoint_refine": true, |
| "mask_downsample_ratio": 4, |
| "num_classes": 1, |
| "num_queries": 300, |
| "num_select": 300, |
| "num_windows": 2, |
| "out_feature_indexes": [ |
| 3, |
| 6, |
| 9, |
| 12 |
| ], |
| "patch_size": 12, |
| "positional_encoding_size": 64, |
| "projector_scale": [ |
| "P4" |
| ], |
| "resolution": 768, |
| "sa_nheads": 8, |
| "segmentation_head": true, |
| "two_stage": true |
| }, |
| "proposal_policy": { |
| "mask_threshold": 0.5, |
| "max_regions": 100, |
| "nms_threshold": 0.6, |
| "reject_empty_masks": true, |
| "score_threshold": 0.15, |
| "version": "falcon-proposals/v1" |
| }, |
| "variant": "seg-2xlarge" |
| }, |
| "image_processor_config": { |
| "_processor_class": null, |
| "crop_size": { |
| "height": 448, |
| "width": 448 |
| }, |
| "do_center_crop": true, |
| "do_convert_rgb": true, |
| "do_normalize": true, |
| "do_rescale": true, |
| "do_resize": true, |
| "image_mean": [ |
| 0.485, |
| 0.456, |
| 0.406 |
| ], |
| "image_processor_type": "BitImageProcessor", |
| "image_std": [ |
| 0.229, |
| 0.224, |
| 0.225 |
| ], |
| "resample": 3, |
| "rescale_factor": 0.00392156862745098, |
| "size": { |
| "height": 448, |
| "width": 448 |
| } |
| }, |
| "language_config": { |
| "_attn_implementation_autoset": true, |
| "add_cross_attention": false, |
| "architectures": [ |
| "LlamaForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "attn_implementation": "sdpa", |
| "bad_words_ids": null, |
| "begin_suppress_tokens": null, |
| "bos_token_id": 1, |
| "chunk_size_feed_forward": 0, |
| "cross_attention_hidden_size": null, |
| "decoder_start_token_id": null, |
| "diversity_penalty": 0.0, |
| "do_sample": false, |
| "early_stopping": false, |
| "encoder_no_repeat_ngram_size": 0, |
| "eos_token_id": 2, |
| "exponential_decay_length_penalty": null, |
| "finetuning_task": null, |
| "forced_bos_token_id": null, |
| "forced_eos_token_id": null, |
| "head_dim": 128, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "id2label": { |
| "0": "LABEL_0", |
| "1": "LABEL_1" |
| }, |
| "initializer_range": 0.02, |
| "intermediate_size": 11008, |
| "is_decoder": false, |
| "is_encoder_decoder": false, |
| "label2id": { |
| "LABEL_0": 0, |
| "LABEL_1": 1 |
| }, |
| "length_penalty": 1.0, |
| "max_length": 20, |
| "max_position_embeddings": 4096, |
| "min_length": 0, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "no_repeat_ngram_size": 0, |
| "num_attention_heads": 32, |
| "num_beam_groups": 1, |
| "num_beams": 1, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 32, |
| "num_return_sequences": 1, |
| "output_attentions": false, |
| "output_hidden_states": false, |
| "output_scores": false, |
| "pad_token_id": 0, |
| "prefix": null, |
| "pretraining_tp": 1, |
| "problem_type": null, |
| "pruned_heads": {}, |
| "remove_invalid_values": false, |
| "repetition_penalty": 1.0, |
| "return_dict": true, |
| "return_dict_in_generate": false, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": null, |
| "rope_theta": 10000.0, |
| "sep_token_id": null, |
| "suppress_tokens": null, |
| "task_specific_params": null, |
| "temperature": 1.0, |
| "tf_legacy_loss": false, |
| "tie_encoder_decoder": false, |
| "tie_word_embeddings": false, |
| "tokenizer_class": null, |
| "top_k": 50, |
| "top_p": 1.0, |
| "torch_dtype": "bfloat16", |
| "torchscript": false, |
| "transformers_version": "4.49.0", |
| "typical_p": 1.0, |
| "use_bfloat16": false, |
| "use_cache": true, |
| "vocab_size": 32000 |
| }, |
| "language_generation_config": { |
| "_from_model_config": false, |
| "assistant_confidence_threshold": 0.4, |
| "assistant_early_exit": null, |
| "assistant_lookbehind": 10, |
| "bad_words_ids": null, |
| "begin_suppress_tokens": null, |
| "bos_token_id": 1, |
| "cache_config": null, |
| "cache_implementation": null, |
| "constraints": null, |
| "decoder_start_token_id": null, |
| "disable_compile": false, |
| "diversity_penalty": 0.0, |
| "do_sample": false, |
| "dola_layers": null, |
| "early_stopping": false, |
| "encoder_no_repeat_ngram_size": 0, |
| "encoder_repetition_penalty": 1.0, |
| "eos_token_id": 2, |
| "epsilon_cutoff": 0.0, |
| "eta_cutoff": 0.0, |
| "exponential_decay_length_penalty": null, |
| "force_words_ids": null, |
| "forced_bos_token_id": null, |
| "forced_decoder_ids": null, |
| "forced_eos_token_id": null, |
| "generation_kwargs": {}, |
| "guidance_scale": null, |
| "is_assistant": false, |
| "length_penalty": 1.0, |
| "low_memory": null, |
| "max_length": 4096, |
| "max_matching_ngram_size": null, |
| "max_new_tokens": null, |
| "max_time": null, |
| "min_length": 0, |
| "min_new_tokens": null, |
| "min_p": null, |
| "no_repeat_ngram_size": 0, |
| "num_assistant_tokens": 20, |
| "num_assistant_tokens_schedule": "constant", |
| "num_beam_groups": 1, |
| "num_beams": 1, |
| "num_return_sequences": 1, |
| "output_attentions": false, |
| "output_hidden_states": false, |
| "output_logits": null, |
| "output_scores": false, |
| "pad_token_id": 0, |
| "penalty_alpha": null, |
| "prompt_lookup_num_tokens": null, |
| "remove_invalid_values": false, |
| "renormalize_logits": false, |
| "repetition_penalty": 1.0, |
| "return_dict_in_generate": false, |
| "return_legacy_cache": null, |
| "sequence_bias": null, |
| "stop_strings": null, |
| "suppress_tokens": null, |
| "target_lookbehind": 10, |
| "temperature": 0.9, |
| "token_healing": false, |
| "top_k": 50, |
| "top_p": 0.6, |
| "transformers_version": "4.49.0", |
| "typical_p": 1.0, |
| "use_cache": true, |
| "watermarking_config": null |
| }, |
| "lora_config": { |
| "alpha": 32, |
| "dropout": 0.05, |
| "rank": 16, |
| "target_modules": [ |
| "k_proj", |
| "o_proj", |
| "q_proj", |
| "v_proj" |
| ] |
| }, |
| "model_type": "falcon_x", |
| "runtime_config": { |
| "detector": { |
| "mask_threshold": 0.5, |
| "max_regions": 100, |
| "nms_threshold": 0.6, |
| "resolution": 768, |
| "score_threshold": 0.15, |
| "variant": "seg-2xlarge" |
| }, |
| "experiment": { |
| "backbone_layout": "independent", |
| "name": "falcon-x", |
| "protocol": "dataset-v1", |
| "ssa_ablation": "none", |
| "world_size": null |
| }, |
| "model": { |
| "component_order": [ |
| "detonator", |
| "explosive", |
| "battery" |
| ], |
| "image_size": 448, |
| "language_model": "lmsys/vicuna-7b-v1.5", |
| "link_order": [ |
| [ |
| "battery", |
| "detonator" |
| ], |
| [ |
| "battery", |
| "explosive" |
| ], |
| [ |
| "detonator", |
| "explosive" |
| ] |
| ], |
| "lora_alpha": 32, |
| "lora_dropout": 0.05, |
| "lora_rank": 16, |
| "max_regions": 100, |
| "patch_size": 14, |
| "region_dim": 1024, |
| "roi_size": 4, |
| "vision_model": "facebook/dinov2-large" |
| }, |
| "stages": { |
| "stage1": { |
| "batch_size": 1, |
| "encoder_learning_rate": 0.00015, |
| "epochs": 12, |
| "expanded_scales": false, |
| "gradient_accumulation": 16, |
| "learning_rate": 0.0001, |
| "lr_scheduler": "cosine", |
| "multi_scale": false, |
| "precision": "backend_mixed", |
| "tf32": true, |
| "warmup_epochs": 0.0, |
| "weight_decay": 0.0001 |
| }, |
| "stage2": { |
| "batch_size": 4, |
| "epoch_size": null, |
| "epochs": 1, |
| "gradient_accumulation": 4, |
| "learning_rate": 0.0001, |
| "link_loss_weight": 0.5, |
| "max_text_tokens": 256, |
| "precision": "bf16", |
| "presence_loss_weight": 0.5, |
| "risk_loss_weight": 1.0, |
| "sampling": "all", |
| "seed": 42, |
| "text_overflow_policy": "error", |
| "tf32": true, |
| "warmup_ratio": 0.03, |
| "weight_decay": 0.0 |
| }, |
| "stage3": { |
| "batch_size": 1, |
| "epoch_size": null, |
| "epochs": 1, |
| "gradient_accumulation": 16, |
| "learning_rate": 1e-05, |
| "link_loss_weight": 0.5, |
| "max_text_tokens": 256, |
| "precision": "bf16", |
| "presence_loss_weight": 0.5, |
| "risk_loss_weight": 1.0, |
| "sampling": "all", |
| "seed": 42, |
| "text_overflow_policy": "error", |
| "tf32": true, |
| "warmup_ratio": 0.03, |
| "weight_decay": 0.0 |
| } |
| }, |
| "training": { |
| "batch_size": 1, |
| "epoch_size": null, |
| "epochs": 1, |
| "gradient_accumulation": 16, |
| "learning_rate": 0.0001, |
| "link_loss_weight": 0.5, |
| "max_text_tokens": 256, |
| "precision": "bf16", |
| "presence_loss_weight": 0.5, |
| "risk_loss_weight": 1.0, |
| "sampling": "all", |
| "seed": 42, |
| "text_overflow_policy": "error", |
| "tf32": true, |
| "warmup_ratio": 0.03, |
| "weight_decay": 0.0 |
| } |
| }, |
| "tie_word_embeddings": false, |
| "torch_dtype": "float32", |
| "transformers_version": "4.49.0", |
| "vision_config": { |
| "_attn_implementation_autoset": true, |
| "add_cross_attention": false, |
| "apply_layernorm": true, |
| "architectures": [ |
| "Dinov2Model" |
| ], |
| "attention_probs_dropout_prob": 0.0, |
| "attn_implementation": "sdpa", |
| "bad_words_ids": null, |
| "begin_suppress_tokens": null, |
| "bos_token_id": null, |
| "chunk_size_feed_forward": 0, |
| "cross_attention_hidden_size": null, |
| "decoder_start_token_id": null, |
| "diversity_penalty": 0.0, |
| "do_sample": false, |
| "drop_path_rate": 0.0, |
| "early_stopping": false, |
| "encoder_no_repeat_ngram_size": 0, |
| "eos_token_id": null, |
| "exponential_decay_length_penalty": null, |
| "finetuning_task": null, |
| "forced_bos_token_id": null, |
| "forced_eos_token_id": null, |
| "hidden_act": "gelu", |
| "hidden_dropout_prob": 0.0, |
| "hidden_size": 1024, |
| "id2label": { |
| "0": "LABEL_0", |
| "1": "LABEL_1" |
| }, |
| "image_size": 518, |
| "initializer_range": 0.02, |
| "is_decoder": false, |
| "is_encoder_decoder": false, |
| "label2id": { |
| "LABEL_0": 0, |
| "LABEL_1": 1 |
| }, |
| "layer_norm_eps": 1e-06, |
| "layerscale_value": 1.0, |
| "length_penalty": 1.0, |
| "max_length": 20, |
| "min_length": 0, |
| "mlp_ratio": 4, |
| "model_type": "dinov2", |
| "no_repeat_ngram_size": 0, |
| "num_attention_heads": 16, |
| "num_beam_groups": 1, |
| "num_beams": 1, |
| "num_channels": 3, |
| "num_hidden_layers": 24, |
| "num_return_sequences": 1, |
| "out_features": [ |
| "stage24" |
| ], |
| "out_indices": [ |
| 24 |
| ], |
| "output_attentions": false, |
| "output_hidden_states": false, |
| "output_scores": false, |
| "pad_token_id": null, |
| "patch_size": 14, |
| "prefix": null, |
| "problem_type": null, |
| "pruned_heads": {}, |
| "qkv_bias": true, |
| "remove_invalid_values": false, |
| "repetition_penalty": 1.0, |
| "reshape_hidden_states": true, |
| "return_dict": true, |
| "return_dict_in_generate": false, |
| "sep_token_id": null, |
| "stage_names": [ |
| "stem", |
| "stage1", |
| "stage2", |
| "stage3", |
| "stage4", |
| "stage5", |
| "stage6", |
| "stage7", |
| "stage8", |
| "stage9", |
| "stage10", |
| "stage11", |
| "stage12", |
| "stage13", |
| "stage14", |
| "stage15", |
| "stage16", |
| "stage17", |
| "stage18", |
| "stage19", |
| "stage20", |
| "stage21", |
| "stage22", |
| "stage23", |
| "stage24" |
| ], |
| "suppress_tokens": null, |
| "task_specific_params": null, |
| "temperature": 1.0, |
| "tf_legacy_loss": false, |
| "tie_encoder_decoder": false, |
| "tie_word_embeddings": true, |
| "tokenizer_class": null, |
| "top_k": 50, |
| "top_p": 1.0, |
| "torch_dtype": "float32", |
| "torchscript": false, |
| "transformers_version": "4.49.0", |
| "typical_p": 1.0, |
| "use_bfloat16": false, |
| "use_mask_token": true, |
| "use_swiglu_ffn": false |
| } |
| } |
|
|
