在 __wct_core 中,当 cont_feat 或 styl_feat 为 [x,1] 维矩阵时, SVD 失败
作者: dkreinov创建于 2020年7月19日更新于 2021年11月3日
除非我做错了什么,在 **def __wct_core** 中存在一个特殊情况,当其中一个矩阵基本上是一个向量时。在计算时: **contentConv = torch.mm(cont_feat, cont_feat.t()).div(cFSize[1] - 1) + iden** **cFSize[1] 为 1**,因此存在除以 0 的情况 => 我们得到一个全是 NAN 的矩阵,导致 SVD 失败。
内容来源: NVIDIA/FastPhotoStyle