[Error] StyleGAN2 | AssertionError : assert state["version"] in [2, 3, 4]

오늘도 에러 사냥

.

https://github.com/mit-han-lab/data-efficient-gans
DataEfficientGans로 transfer-learning해보다가 난 에러..
요 자체에서 나는 에러보다 stylegan2 tf에서 나는 에러같다. stylegan2 ada tf도 동일하다고 함.

AssertionError : assert state["version"] in [2, 3, 4]

• 뭔가 버전이 맞지 않는다고 한다.

• 에러가 난 코드를 찾아가 print를 찍어본다
• 난 도대체 버전이 몇이길래..

• 버전이 5라고 한다

# assert state["version"] in [2, 3, 4]

• assert를 무시해버리면?

TypeError: fused_bias_act() got an unexpected keyword argument 'clamp'

• 될 리가 없다

• 체크포인트들이 version 5이고
• version 5 fused_bias는 적어도 하나 이상의 인수 "clamp"를 가지고 있음

해결

내가 사용한 pretrained 모델이 stylegan2-ada 모델인데 위 레포는 stylegan2 모델만 support해주기 때문에 난 에러였다.
pretrained 모델을 stylegan2 모델로 변경하여 해결했다.

---

References

https://github.com/rosinality/stylegan2-pytorch/issues/164