Skip to content

Commit 0d4ea12

Browse files
authored
[New Sample] Add Some zero-shot-object-detection Model Computational Graph (#203)
1 parent 3588050 commit 0d4ea12

File tree

78 files changed

+85024
-0
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

78 files changed

+85024
-0
lines changed
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
affc8c7cd95d8acc9ee5582af79eef03a74c19d6a08a54cd80ec10a6086cdba1
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": false
6+
}

samples/transformers-auto-model/IDEA-Research_grounding-dino-base/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/IDEA-Research_grounding-dino-base/input_tensor_constraints.py

Whitespace-only changes.

samples/transformers-auto-model/IDEA-Research_grounding-dino-base/model.py

Lines changed: 363 additions & 0 deletions
Large diffs are not rendered by default.
Lines changed: 114 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,114 @@
1+
class Program_weight_tensor_meta_L_stack0_encoder_last_hidden_state_text:
2+
name = "L_stack0_encoder_last_hidden_state_text"
3+
shape = [1, 7, 256]
4+
dtype = "torch.float32"
5+
device = "cuda:0"
6+
mean = 0.000
7+
std = 1.000
8+
data = None
9+
10+
11+
class Program_weight_tensor_meta_L_stack0_intermediate_hidden_states:
12+
name = "L_stack0_intermediate_hidden_states"
13+
shape = [1, 6, 900, 256]
14+
dtype = "torch.float32"
15+
device = "cuda:0"
16+
mean = 0.000
17+
std = 1.000
18+
data = None
19+
20+
21+
class Program_weight_tensor_meta_L_stack0_init_reference_points:
22+
name = "L_stack0_init_reference_points"
23+
shape = [1, 900, 4]
24+
dtype = "torch.float32"
25+
device = "cuda:0"
26+
mean = 0.400
27+
std = 0.296
28+
data = None
29+
30+
31+
class Program_weight_tensor_meta_L_stack0_intermediate_reference_points:
32+
name = "L_stack0_intermediate_reference_points"
33+
shape = [1, 6, 900, 4]
34+
dtype = "torch.float32"
35+
device = "cuda:0"
36+
mean = 0.400
37+
std = 0.296
38+
data = None
39+
40+
41+
class Program_weight_tensor_meta_L_attention_mask_:
42+
name = "L_attention_mask_"
43+
shape = [1, 7]
44+
dtype = "torch.int64"
45+
device = "cuda:0"
46+
mean = None
47+
std = None
48+
data = [1, 1, 1, 1, 1, 1, 1]
49+
50+
51+
class Program_weight_tensor_meta_L_self_modules_bbox_embed_modules_0_modules_layers_modules_0_parameters_weight_:
52+
name = "L_self_modules_bbox_embed_modules_0_modules_layers_modules_0_parameters_weight_"
53+
shape = [256, 256]
54+
dtype = "torch.float32"
55+
device = "cuda:0"
56+
mean = -0.000
57+
std = 0.020
58+
data = None
59+
60+
61+
class Program_weight_tensor_meta_L_self_modules_bbox_embed_modules_0_modules_layers_modules_0_parameters_bias_:
62+
name = (
63+
"L_self_modules_bbox_embed_modules_0_modules_layers_modules_0_parameters_bias_"
64+
)
65+
shape = [256]
66+
dtype = "torch.float32"
67+
device = "cuda:0"
68+
mean = 0.000
69+
std = 0.000
70+
data = None
71+
72+
73+
class Program_weight_tensor_meta_L_self_modules_bbox_embed_modules_0_modules_layers_modules_1_parameters_weight_:
74+
name = "L_self_modules_bbox_embed_modules_0_modules_layers_modules_1_parameters_weight_"
75+
shape = [256, 256]
76+
dtype = "torch.float32"
77+
device = "cuda:0"
78+
mean = 0.000
79+
std = 0.020
80+
data = None
81+
82+
83+
class Program_weight_tensor_meta_L_self_modules_bbox_embed_modules_0_modules_layers_modules_1_parameters_bias_:
84+
name = (
85+
"L_self_modules_bbox_embed_modules_0_modules_layers_modules_1_parameters_bias_"
86+
)
87+
shape = [256]
88+
dtype = "torch.float32"
89+
device = "cuda:0"
90+
mean = 0.000
91+
std = 0.000
92+
data = None
93+
94+
95+
class Program_weight_tensor_meta_L_self_modules_bbox_embed_modules_0_modules_layers_modules_2_parameters_weight_:
96+
name = "L_self_modules_bbox_embed_modules_0_modules_layers_modules_2_parameters_weight_"
97+
shape = [4, 256]
98+
dtype = "torch.float32"
99+
device = "cuda:0"
100+
mean = 0.000
101+
std = 0.000
102+
data = None
103+
104+
105+
class Program_weight_tensor_meta_L_self_modules_bbox_embed_modules_0_modules_layers_modules_2_parameters_bias_:
106+
name = (
107+
"L_self_modules_bbox_embed_modules_0_modules_layers_modules_2_parameters_bias_"
108+
)
109+
shape = [4]
110+
dtype = "torch.float32"
111+
device = "cuda:0"
112+
mean = 0.000
113+
std = 0.000
114+
data = [0.000000, 0.000000, 0.000000, 0.000000]
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
3eb7ef4fc44c784bf246435c08ca58a7f6defb71f4707d5dfa27e4bf629fc09d
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"framework": "torch",
3+
"num_devices_required": 1,
4+
"num_nodes_required": 1,
5+
"dynamic": true
6+
}

samples/transformers-auto-model/IDEA-Research_grounding-dino-tiny/input_meta.py

Whitespace-only changes.

samples/transformers-auto-model/IDEA-Research_grounding-dino-tiny/input_tensor_constraints.py

Whitespace-only changes.

0 commit comments

Comments
 (0)