fixes here and there

train.py

@@ -33,7 +33,7 @@ def __init__(self, config, train: DataLoader, val: DataLoader):
 
     def train(self):
         self._init_params()
-        for epoch in range(0, config['num_epochs']):
+        for epoch in range(0, self.config['num_epochs']):
             if (epoch == self.warmup_epochs) and not (self.warmup_epochs == 0):
                 self.netG.module.unfreeze()
                 self.optimizer_G = self._get_optim(self.netG.parameters())