search

seanshpark / help

common help
MIT License
4 stars 0 forks source link

Python #16

Open seanshpark opened 2 years ago

seanshpark commented 2 years ago

Understanding python;

seanshpark commented 2 years ago 
var = 'hello'
print(f'message {var}')

-->

message hello
seanshpark commented 2 years ago

Exporting with dynamic shape

torch.onnx.export(model, inputs, "model.onnx", export_params=True,
                  input_names=input_names, output_names=output_names,
                  dynamic_axes={"input":[0]})