Spaces:

gyrojeff
/

YuzuMarker.FontDetection

Running

gyrojeff commited on Mar 29, 2023

Commit

e980b66

1 Parent(s): 9fb34ba

fix: typo

Files changed (1) hide show

detector/model.py CHANGED Viewed

@@ -158,7 +158,7 @@ class FontDetector(ptl.LightningModule):
         self.num_warmup_iters = num_warmup_iters
         self.num_iters = num_iters
         self.num_epochs = num_epochs
-        self.load_step = 0
     def forward(self, x):
         return self.model(x)

         self.num_warmup_iters = num_warmup_iters
         self.num_iters = num_iters
         self.num_epochs = num_epochs
+        self.load_epoch = 0
     def forward(self, x):
         return self.model(x)