torch.onnx.export(
  mbr_core,
  torch.rand(1, 801, 60, 384, dtype=torch.float32),
  "model.onnx",
  export_params=True,
  input_names = ['input'],
  output_names = ['output'],
  dynamic_axes = {
    'input' : {1 : 'duration'},
    'output' : {1 : 'duration'}
  }
)

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for mochiya98/melbandroformer_variant

Base model

KimberleyJSN/melbandroformer

Quantized

(1)

this model