onnx_chainer.export_testcase¶
-
onnx_chainer.
export_testcase
(model, args, out_dir, output_grad=False, **kwargs)[source]¶ Export model and I/O tensors of the model in protobuf format.
Similar to the export function, this function first performs a forward computation to a given input for obtaining an output. Then, this function saves the pair of input and output in Protobuf format, which is a defacto-standard format in ONNX.
This function also saves the model with the name “model.onnx”.
Parameters: - model (Chain) – The model object.
- args (list) – The arguments which are given to the model directly. Unlike export function, only list type is accepted.
- out_dir (str) – The directory name used for saving the input and output.
- output_grad (bool) – If True, this function will output model’s gradient with names ‘gradient_%d.pb’.
- **kwargs (dict) – keyword arguments for
onnx_chainer.export
.