debug

README.md

@@ -1,5 +1,5 @@
 ---
-title: OptVQ
+title: ReVQ
 emoji: 😻
 colorFrom: indigo
 colorTo: red

revq/losses/aeloss_disc.py

@@ -15,7 +15,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 import lpips
 
-from optvq.models.discriminator import NLayerDiscriminator, weights_init
+from revq.models.discriminator import NLayerDiscriminator, weights_init
 
 class DummyLoss(nn.Module):
     def __init__(self):

revq/models/vqgan.py

@@ -11,7 +11,7 @@ import torch
 import torch.nn.functional as F
 import torch.nn as nn
 
-import optvq.utils.logger as L
+import revq.utils.logger as L
 
 class Identity(nn.Module):
     def forward(self, x):

revq/trainer/pipeline.py

@@ -15,9 +15,9 @@ import torch
 import torch.nn as nn
 from torch.utils.data.dataloader import DataLoader
 
-from optvq.utils.init import initiate_from_config_recursively
-from optvq.data.dataloader import maybe_get_subset
-import optvq.utils.logger as L
+from revq.utils.init import initiate_from_config_recursively
+from revq.data.dataloader import maybe_get_subset
+import revq.utils.logger as L
 
 def setup_config(opt: argparse.Namespace):
     L.log.info("\n\n### Setting up the configurations. ###")

revq/utils/logger.py

@@ -25,9 +25,9 @@ except:
 
 """
 NOTE: The `log` instance is a global variable, which should be imported by other modules as:
-    `import optvq.utils.logger as logger` 
+    `import revq.utils.logger as logger` 
 rather than 
-    `from optvq.utils.logger import log`.
+    `from revq.utils.logger import log`.
 """
 
 def setup_printer(file_log_dir: str, use_console: bool = True):