update

2020-09-05 22:00:17 +08:00
parent 39c754374c
commit e3c760d0c5
12 changed files with 122 additions and 43 deletions
--- a/model/GAN/UGATIT.py
+++ b/model/GAN/UGATIT.py
@@ -1,6 +1,6 @@
 import torch
 import torch.nn as nn
-from .residual_generator import ResidualBlock
+from .base import ResidualBlock
 from model.registry import MODEL