<p dir="auto">when I use:<br>
torch.onnx.export()<br>
to convert my pytorch model to onnx model,<br>
I can monitor the memory is used up by the converting process.<br>
and then the process is killed.<br>
no any other logs, just killed.</p>
<p dir="auto">cc <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/houseroad/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/houseroad">@houseroad</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/spandantiwari/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/spandantiwari">@spandantiwari</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/lara-hdr/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/lara-hdr">@lara-hdr</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/BowenBao/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/BowenBao">@BowenBao</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/neginraoof/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/neginraoof">@neginraoof</a></p>
<h2 dir="auto">🐛 Bug</h2>
<p dir="auto">See also <a class="issue-link js-issue-link" data-error-text="Failed to load title" data-id="408355158" data-permission-text="Title is private" data-url="https://github.com/pytorch/pytorch/issues/16908" data-hovercard-type="issue" data-hovercard-url="/pytorch/pytorch/issues/16908/hovercard" href="https://github.com/pytorch/pytorch/issues/16908">#16908</a><br>
After fixing INT64 issue we moved forward and stumbled upon the following:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="ERROR:  In node 175 (convert_axis): UNSUPPORTED_NODE: Assertion failed: axis &gt;= 0 &amp;&amp; axis &lt; nbDims"><pre class="notranslate"><code class="notranslate">ERROR:  In node 175 (convert_axis): UNSUPPORTED_NODE: Assertion failed: axis &gt;= 0 &amp;&amp; axis &lt; nbDims
</code></pre></div>
<p dir="auto">This is Gather on Axis 0. Seems like it has to be another axis. Please check it out.</p>