example_outputs=None, strip_doc_string=True, dynamic_axes=None, keep_initializers_as_inputs=None)
功能:
将模型以...参数:
model (torch.nn.Module) – 要导出的模型.
args (tuple of arguments) – 模型的输入, 任何非Tensor参数都将硬编码到导出的模型中;任何Tensor...因此,导出的模型将接受大小为[batch_size,3、100、100]的输入,其中batch_size可以是可变的。...如果你要导出一个没训练过的就设为 False.
verbose (bool, default False) - 如果指定,我们将打印出一个导出轨迹的调试描述。...None, output_names=None, dynamic_axes = {‘input_1’:[0, 2, 3], ‘input_2’:[0], ‘output’:[0, 1]})
你好,请问pytorch