Spaces:

ankankbhunia
/

HWT

Running

ankankbhunia commited on Feb 20, 2024

Commit

3196ac3

verified ·

1 Parent(s): 6613ed6

Update models/model.py

Files changed (1) hide show

models/model.py CHANGED Viewed

@@ -316,7 +316,7 @@ class TRGAN(nn.Module):
         self.netOCR = CRNN().to(DEVICE)
         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.netOCR = CRNN().to(DEVICE)
         self.OCR_criterion = CTCLoss(zero_infinity=True, reduction='none')
+        #block_idx = InceptionV3.BLOCK_INDEX_BY_DIM[2048]
         #self.inception = InceptionV3([block_idx]).to(DEVICE)