Francesco commited on
Commit
1ee73bd
1 Parent(s): ab01c75
Files changed (2) hide show
  1. config.json +92 -92
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -96,6 +96,97 @@
96
  },
97
  "ce_weight": 1.0,
98
  "cross_entropy_weight": 1.0,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
99
  "dice_weight": 1.0,
100
  "fpn_feature_size": 256,
101
  "id2label": {
@@ -378,97 +469,6 @@
378
  "num_hidden_layers": 6,
379
  "num_queries": 100,
380
  "torch_dtype": "float32",
381
- "transformer_decoder_config": {
382
- "_name_or_path": "",
383
- "activation_dropout": 0.0,
384
- "activation_function": "relu",
385
- "add_cross_attention": false,
386
- "architectures": null,
387
- "attention_dropout": 0.0,
388
- "auxiliary_loss": false,
389
- "backbone": "resnet50",
390
- "bad_words_ids": null,
391
- "bbox_cost": 5,
392
- "bbox_loss_coefficient": 5,
393
- "bos_token_id": null,
394
- "chunk_size_feed_forward": 0,
395
- "class_cost": 1,
396
- "cross_attention_hidden_size": null,
397
- "d_model": 256,
398
- "decoder_attention_heads": 8,
399
- "decoder_ffn_dim": 2048,
400
- "decoder_layerdrop": 0.0,
401
- "decoder_layers": 6,
402
- "decoder_start_token_id": null,
403
- "dice_loss_coefficient": 1,
404
- "dilation": false,
405
- "diversity_penalty": 0.0,
406
- "do_sample": false,
407
- "dropout": 0.1,
408
- "early_stopping": false,
409
- "encoder_attention_heads": 8,
410
- "encoder_ffn_dim": 2048,
411
- "encoder_layerdrop": 0.0,
412
- "encoder_layers": 6,
413
- "encoder_no_repeat_ngram_size": 0,
414
- "eos_coefficient": 0.1,
415
- "eos_token_id": null,
416
- "finetuning_task": null,
417
- "forced_bos_token_id": null,
418
- "forced_eos_token_id": null,
419
- "giou_cost": 2,
420
- "giou_loss_coefficient": 2,
421
- "id2label": {
422
- "0": "LABEL_0",
423
- "1": "LABEL_1"
424
- },
425
- "init_std": 0.02,
426
- "init_xavier_std": 1.0,
427
- "is_decoder": false,
428
- "is_encoder_decoder": true,
429
- "label2id": {
430
- "LABEL_0": 0,
431
- "LABEL_1": 1
432
- },
433
- "length_penalty": 1.0,
434
- "mask_loss_coefficient": 1,
435
- "max_length": 20,
436
- "max_position_embeddings": 1024,
437
- "min_length": 0,
438
- "model_type": "detr",
439
- "no_repeat_ngram_size": 0,
440
- "num_beam_groups": 1,
441
- "num_beams": 1,
442
- "num_hidden_layers": 6,
443
- "num_queries": 100,
444
- "num_return_sequences": 1,
445
- "output_attentions": false,
446
- "output_hidden_states": false,
447
- "output_scores": false,
448
- "pad_token_id": null,
449
- "position_embedding_type": "sine",
450
- "prefix": null,
451
- "problem_type": null,
452
- "pruned_heads": {},
453
- "remove_invalid_values": false,
454
- "repetition_penalty": 1.0,
455
- "return_dict": true,
456
- "return_dict_in_generate": false,
457
- "scale_embedding": false,
458
- "sep_token_id": null,
459
- "task_specific_params": null,
460
- "temperature": 1.0,
461
- "tie_encoder_decoder": false,
462
- "tie_word_embeddings": true,
463
- "tokenizer_class": null,
464
- "top_k": 50,
465
- "top_p": 1.0,
466
- "torch_dtype": null,
467
- "torchscript": false,
468
- "transformers_version": "4.17.0.dev0",
469
- "typical_p": 1.0,
470
- "use_bfloat16": false
471
- },
472
  "transformers_version": null,
473
- "use_auxilary_loss": false
474
  }
 
96
  },
97
  "ce_weight": 1.0,
98
  "cross_entropy_weight": 1.0,
99
+ "decoder_config": {
100
+ "_name_or_path": "",
101
+ "activation_dropout": 0.0,
102
+ "activation_function": "relu",
103
+ "add_cross_attention": false,
104
+ "architectures": null,
105
+ "attention_dropout": 0.0,
106
+ "auxiliary_loss": false,
107
+ "backbone": "resnet50",
108
+ "bad_words_ids": null,
109
+ "bbox_cost": 5,
110
+ "bbox_loss_coefficient": 5,
111
+ "bos_token_id": null,
112
+ "chunk_size_feed_forward": 0,
113
+ "class_cost": 1,
114
+ "cross_attention_hidden_size": null,
115
+ "d_model": 256,
116
+ "decoder_attention_heads": 8,
117
+ "decoder_ffn_dim": 2048,
118
+ "decoder_layerdrop": 0.0,
119
+ "decoder_layers": 6,
120
+ "decoder_start_token_id": null,
121
+ "dice_loss_coefficient": 1,
122
+ "dilation": false,
123
+ "diversity_penalty": 0.0,
124
+ "do_sample": false,
125
+ "dropout": 0.1,
126
+ "early_stopping": false,
127
+ "encoder_attention_heads": 8,
128
+ "encoder_ffn_dim": 2048,
129
+ "encoder_layerdrop": 0.0,
130
+ "encoder_layers": 6,
131
+ "encoder_no_repeat_ngram_size": 0,
132
+ "eos_coefficient": 0.1,
133
+ "eos_token_id": null,
134
+ "finetuning_task": null,
135
+ "forced_bos_token_id": null,
136
+ "forced_eos_token_id": null,
137
+ "giou_cost": 2,
138
+ "giou_loss_coefficient": 2,
139
+ "id2label": {
140
+ "0": "LABEL_0",
141
+ "1": "LABEL_1"
142
+ },
143
+ "init_std": 0.02,
144
+ "init_xavier_std": 1.0,
145
+ "is_decoder": false,
146
+ "is_encoder_decoder": true,
147
+ "label2id": {
148
+ "LABEL_0": 0,
149
+ "LABEL_1": 1
150
+ },
151
+ "length_penalty": 1.0,
152
+ "mask_loss_coefficient": 1,
153
+ "max_length": 20,
154
+ "max_position_embeddings": 1024,
155
+ "min_length": 0,
156
+ "model_type": "detr",
157
+ "no_repeat_ngram_size": 0,
158
+ "num_beam_groups": 1,
159
+ "num_beams": 1,
160
+ "num_hidden_layers": 6,
161
+ "num_queries": 100,
162
+ "num_return_sequences": 1,
163
+ "output_attentions": false,
164
+ "output_hidden_states": false,
165
+ "output_scores": false,
166
+ "pad_token_id": null,
167
+ "position_embedding_type": "sine",
168
+ "prefix": null,
169
+ "problem_type": null,
170
+ "pruned_heads": {},
171
+ "remove_invalid_values": false,
172
+ "repetition_penalty": 1.0,
173
+ "return_dict": true,
174
+ "return_dict_in_generate": false,
175
+ "scale_embedding": false,
176
+ "sep_token_id": null,
177
+ "task_specific_params": null,
178
+ "temperature": 1.0,
179
+ "tie_encoder_decoder": false,
180
+ "tie_word_embeddings": true,
181
+ "tokenizer_class": null,
182
+ "top_k": 50,
183
+ "top_p": 1.0,
184
+ "torch_dtype": null,
185
+ "torchscript": false,
186
+ "transformers_version": "4.17.0.dev0",
187
+ "typical_p": 1.0,
188
+ "use_bfloat16": false
189
+ },
190
  "dice_weight": 1.0,
191
  "fpn_feature_size": 256,
192
  "id2label": {
 
469
  "num_hidden_layers": 6,
470
  "num_queries": 100,
471
  "torch_dtype": "float32",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
472
  "transformers_version": null,
473
+ "use_auxiliary_loss": false
474
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f7687e2d0c76f7612fa745ee51400cef5c9d723b8a807b0e662aa7831b884713
3
- size 253044849
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e281b6ab65a335e0a30e13b7a0ae9490f353c16a10af2a7c013bb9b9dcf55970
3
+ size 253041713