HWT

Sleeping

ankankbhunia commited on Feb 20, 2024

Commit

6613ed6

verified ·

1 Parent(s): 1129c08

Update models/model.py

Files changed (1) hide show

models/model.py CHANGED Viewed

@@ -317,7 +317,7 @@ class TRGAN(nn.Module):
         self.OCR_criterion = CTCLoss(zero_infinity=True, reduction='none')
         block_idx = InceptionV3.BLOCK_INDEX_BY_DIM[2048]
-        self.inception = InceptionV3([block_idx]).to(DEVICE)
         self.optimizer_G = torch.optim.Adam(self.netG.parameters(),

         self.OCR_criterion = CTCLoss(zero_infinity=True, reduction='none')
         block_idx = InceptionV3.BLOCK_INDEX_BY_DIM[2048]
+        #self.inception = InceptionV3([block_idx]).to(DEVICE)
         self.optimizer_G = torch.optim.Adam(self.netG.parameters(),