gyrojeff commited on
Commit
d358c49
·
1 Parent(s): 69c8e55

fix: add train accur log

Browse files
Files changed (1) hide show
  1. detector/model.py +4 -0
detector/model.py CHANGED
@@ -125,6 +125,10 @@ class FontDetector(ptl.LightningModule):
125
  )
126
 
127
  def on_train_epoch_end(self) -> None:
 
 
 
 
128
  self.font_accur_train.reset()
129
  self.direction_accur_train.reset()
130
 
 
125
  )
126
 
127
  def on_train_epoch_end(self) -> None:
128
+ self.log("train_font_accur", self.font_accur_train.compute(), sync_dist=True)
129
+ self.log(
130
+ "train_direction_accur", self.direction_accur_train.compute(), sync_dist=True
131
+ )
132
  self.font_accur_train.reset()
133
  self.direction_accur_train.reset()
134