TypeError: Unexpected keyword argument passed to optimizer: beta1

使用 python train.py 训练 StyleGAN 时出现如上错误提示。

解决方法:把 train.py 中的 beta1 和 beta2 更改为 beta_1 和 beta_2 即可(与 adam.py 中的描述保持一致)

原因是 StyleGAN 与 tensorflow 2.x 不兼容。所以终极解决办法就是如 StyleGAN GitHub 描述的那样,使用相应版本的 tensorflow、cudatoolkit 和 cudnn。

你可能感兴趣的:(代码,深度学习,人工智能)