Update gpen.py

fixed bug of importing ppgan model
3 years ago · 6092f8d752
parent 21af009be0
commit 6092f8d752
1 changed files with 3 additions and 3 deletions
--- a/paddlers/models/ppgan/models/generators/gpen.py
+++ b/paddlers/models/ppgan/models/generators/gpen.py
@ -17,9 +17,9 @@
 import paddle
 import paddle.nn as nn
 import math
-from ppgan.models.generators import StyleGANv2Generator
-from ppgan.models.discriminators.discriminator_styleganv2 import ConvLayer
-from ppgan.modules.equalized import EqualLinear
+from ...models.generators import StyleGANv2Generator
+from ...models.discriminators.discriminator_styleganv2 import ConvLayer
+from ...modules.equalized import EqualLinear

 class GPEN(nn.Layer):
    def __init__(