csukuangfj commited on
Commit
84e9708
·
1 Parent(s): 11788ff

add models

Browse files
Files changed (2) hide show
  1. export-onnx.py +2 -1
  2. 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:c1b1628a86d12ea9837c2feb0eab3b08ba95111670813d26f09b75b6dbb7540b
3
  size 9512157
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e062cc0a46ece7ccf730418f00320baec7ca9a8322d3d938d9b5778553464429
3
  size 9512157