https://github.com/HSE-LAMBDA/DeepGenerativeModels/blob/spring-2025-DSBA/homeworks/third/vapnev.ipynb