added log interval selector

2024-08-28 12:32:47 -04:00
parent 9a7fa5fa36
commit 9136e1dfe3
3 changed files with 40 additions and 34 deletions
@@ -474,4 +474,8 @@ class HParams:
        return key in self.__dict__

    def __repr__(self):
-        return self.__dict__.__repr__()
+        return self.__dict__.__repr__()
+    
+    def sync_log_interval(self, dataset_length):
+        log_interval = getattr(self,"log_every_epoch",1)
+        self.train.log_interval = int(self.train.batch_size*dataset_length*log_interval)