chairon 2022-04-01 10:11 采纳率: 0%
浏览 590

用自己的数据集训练styleGAN3时报错如下，请问该怎么解决

D:\Code\styleGAN3\stylegan3-main\training\loss.py:56: UserWarning: Specified kernel cache directory could not be created! This disables kernel caching. Specified directory is C:\Users\Administ
RuntimeError: aten::grid_sampler_2d_backward() is missing value for argument 'output_mask'. Declaration: aten::grid_sampler_2d_backward(Tensor grad_output, Tensor input, Tensor grid, int inter
polation_mode, int padding_mode, bool align_corners, bool[2] output_mask) -> (Tensor, Tensor)
PS D:\Code\styleGAN3\stylegan3-main>

  • 写回答

3条回答 默认 最新

  • 不会长胖的斜杠 新星创作者: 后端开发技术领域 2022-04-01 18:13
    关注

    从报错看，代码里面缺少参数output_mask'，可能是你自己的数据集没有进行标签工作
    望采纳

    评论

报告相同问题？

问题事件

  • 创建了问题 4月1日