[오류 해결] RuntimeError: derivative for aten::grid_sampler_2d_backward is not implemented(feat. StyleGAN2-ADA)

StyleGAN2-ADA를 돌려보려고 했는데 다음과 같은 오류가 발생하였다.

 

발생한 오류

aten::grid_sampler_2d_backward 함수에 대한 미분이 StyleGAN2-ada 모델에서 아직 구현되지 않았다는 의미의 오류이다. 

RuntimeError: derivative for aten::grid_sampler_2d_backward is not implemented(feat. StyleGAN2-ada)

 

해결 방법

StyleGAN3의 코드를 가져다 쓰면 된다. 

StyleGAN2-ADA를 clone 했다고 가정했을 때 stylegan2-ada-pytorch/torch_utils/ops 파일 안에 이 두 파일을  StyleGAN3의 파일로 대체하면 된다! 

• conv2d_gradfix.py →  conv2d_gradfix.py
• grid_sample_gradfix.py →  grid_sample_gradfix.py