728x90
반응형
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
728x90
반응형
LIST