|
|
发表于 2026-5-12 20:46:19
|
显示全部楼层
本帖最后由 gaolaodao7788 于 2026-5-21 09:28 编辑
大佬啊,用这个版本跑模型速度非常快,导出SAEHD模型的时候就出问题啦。这个怎么调请指教。拜托
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
正在导出 .dfm 到 D:\MAO-DeepFaceLab-Pytorch\workspace\直播模型\256-WF-D_SAEHD.dfm
[torch.onnx] Obtain model graph for `DFMModel()` with `torch.export.export(..., strict=False)`...
[torch.onnx] Obtain model graph for `DFMModel()` with `torch.export.export(..., strict=False)`... ❌
[torch.onnx] Obtain model graph for `DFMModel()` with `torch.export.export(..., strict=True)`...
[torch.onnx] Obtain model graph for `DFMModel()` with `torch.export.export(..., strict=True)`... ❌
/!\ 导出失败: Failed to export the model with torch.export. This is step 1/3 of exporting the model to ONNX. Next steps:
- Modify the model code for `torch.export.export` to succeed. Refer to https://pytor
Traceback (most recent call last):
File "D:\MAO-DeepFaceLab-Pytorch\_internal\python\Lib\site-packages\torch\onnx\_internal\exporter\_capture_strategies.py", line 121, in __call__
exported_program = self._capture(model, args, kwargs, dynamic_shapes)
File "D:\MAO-DeepFaceLab-Pytorch\_internal\python\Lib\site-packages\torch\onnx\_internal\exporter\_capture_strategies.py", line 219, in _capture
return torch.export.export(
File "D:\MAO-DeepFaceLab-Pytorch\_internal\python\Lib\site-packages\torch\export\__init__.py", line 311, in export
......
......
<class 'RuntimeError'>: shape '[1, 2, 2, 0, 128, 512]' is invalid for input of size 196608
|
|