csukuangfj
commited on
Commit
·
84e9708
1
Parent(s):
11788ff
add models
Browse files- export-onnx.py +2 -1
- model.onnx +1 -1
export-onnx.py
CHANGED
@@ -67,7 +67,8 @@ def main():
|
|
67 |
example_output = model(model.example_input_array)
|
68 |
|
69 |
# (batch, num_frames, num_classes)
|
70 |
-
assert list(example_output.shape) == [1, 589, 7], example_output.shape
|
|
|
71 |
|
72 |
assert model.receptive_field.step == 0.016875, model.receptive_field.step
|
73 |
assert model.receptive_field.duration == 0.0619375, model.receptive_field.duration
|
|
|
67 |
example_output = model(model.example_input_array)
|
68 |
|
69 |
# (batch, num_frames, num_classes)
|
70 |
+
# assert list(example_output.shape) == [1, 589, 7], example_output.shape
|
71 |
+
print(example_output.shape)
|
72 |
|
73 |
assert model.receptive_field.step == 0.016875, model.receptive_field.step
|
74 |
assert model.receptive_field.duration == 0.0619375, model.receptive_field.duration
|
model.onnx
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 9512157
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e062cc0a46ece7ccf730418f00320baec7ca9a8322d3d938d9b5778553464429
|
3 |
size 9512157
|