qubvel-hf HF Staff commited on
Commit
2eba669
·
verified ·
1 Parent(s): 467b575

Training in progress, epoch 1

Browse files
config.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/maskformer-swin-tiny-coco",
3
+ "architectures": [
4
+ "MaskFormerForInstanceSegmentation"
5
+ ],
6
+ "backbone": null,
7
+ "backbone_config": {
8
+ "depths": [
9
+ 2,
10
+ 2,
11
+ 6,
12
+ 2
13
+ ],
14
+ "drop_path_rate": 0.3,
15
+ "embed_dim": 96,
16
+ "encoder_stride": 32,
17
+ "hidden_size": 768,
18
+ "image_size": 224,
19
+ "in_channels": 3,
20
+ "model_type": "maskformer-swin",
21
+ "num_heads": [
22
+ 3,
23
+ 6,
24
+ 12,
25
+ 24
26
+ ],
27
+ "out_features": [
28
+ "stage1",
29
+ "stage2",
30
+ "stage3",
31
+ "stage4"
32
+ ],
33
+ "out_indices": [
34
+ 1,
35
+ 2,
36
+ 3,
37
+ 4
38
+ ],
39
+ "path_norm": true,
40
+ "pretrain_img_size": 224,
41
+ "window_size": 7
42
+ },
43
+ "backbone_kwargs": null,
44
+ "ce_weight": 1.0,
45
+ "cross_entropy_weight": 1.0,
46
+ "decoder_config": {
47
+ "max_position_embeddings": 1024,
48
+ "model_type": "detr",
49
+ "scale_embedding": false
50
+ },
51
+ "dice_weight": 1.0,
52
+ "fpn_feature_size": 256,
53
+ "id2label": {
54
+ "0": "person",
55
+ "1": "car"
56
+ },
57
+ "init_std": 0.02,
58
+ "init_xavier_std": 1.0,
59
+ "label2id": {
60
+ "car": 1,
61
+ "person": 0
62
+ },
63
+ "mask_feature_size": 256,
64
+ "mask_weight": 20.0,
65
+ "model_type": "maskformer",
66
+ "no_object_weight": 0.1,
67
+ "num_attention_heads": 8,
68
+ "num_hidden_layers": 6,
69
+ "num_queries": 100,
70
+ "output_auxiliary_logits": null,
71
+ "torch_dtype": "float32",
72
+ "transformers_version": "4.42.0.dev0",
73
+ "use_auxiliary_loss": false,
74
+ "use_pretrained_backbone": false,
75
+ "use_timm_backbone": false
76
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a386ce688a2c7e1a32f875af96e60c6006d81dc667b1ef170a0441aa6ba6eb2f
3
+ size 167175760
preprocessor_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_max_size": 1333,
3
+ "_valid_processor_keys": [
4
+ "images",
5
+ "segmentation_maps",
6
+ "instance_id_to_semantic_id",
7
+ "do_resize",
8
+ "size",
9
+ "size_divisor",
10
+ "resample",
11
+ "do_rescale",
12
+ "rescale_factor",
13
+ "do_normalize",
14
+ "image_mean",
15
+ "image_std",
16
+ "ignore_index",
17
+ "do_reduce_labels",
18
+ "return_tensors",
19
+ "data_format",
20
+ "input_data_format"
21
+ ],
22
+ "do_normalize": true,
23
+ "do_reduce_labels": true,
24
+ "do_rescale": true,
25
+ "do_resize": true,
26
+ "ignore_index": 255,
27
+ "image_mean": [
28
+ 0.48500001430511475,
29
+ 0.4560000002384186,
30
+ 0.4059999883174896
31
+ ],
32
+ "image_processor_type": "MaskFormerImageProcessor",
33
+ "image_std": [
34
+ 0.2290000021457672,
35
+ 0.2239999920129776,
36
+ 0.22499999403953552
37
+ ],
38
+ "resample": 2,
39
+ "rescale_factor": 0.00392156862745098,
40
+ "size": {
41
+ "height": 256,
42
+ "width": 256
43
+ },
44
+ "size_divisor": 32
45
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:249970984043353f51543f29ce552123fae7b976355e1483fbbc96c9e4ae0356
3
+ size 4731