使用 python train.py 训练 StyleGAN 时出现如上错误提示。
解决方法:把 train.py 中的 beta1 和 beta2 更改为 beta_1 和 beta_2 即可(与 adam.py 中的描述保持一致)
原因是 StyleGAN 与 tensorflow 2.x 不兼容。所以终极解决办法就是如 StyleGAN GitHub 描述的那样,使用相应版本的 tensorflow、cudatoolkit 和 cudnn。
使用 python train.py 训练 StyleGAN 时出现如上错误提示。
解决方法:把 train.py 中的 beta1 和 beta2 更改为 beta_1 和 beta_2 即可(与 adam.py 中的描述保持一致)
原因是 StyleGAN 与 tensorflow 2.x 不兼容。所以终极解决办法就是如 StyleGAN GitHub 描述的那样,使用相应版本的 tensorflow、cudatoolkit 和 cudnn。