From ac1a25096f224d63501d503a486f6129a02bff8c Mon Sep 17 00:00:00 2001
From: lruizcalico <lruiz@calicolabs.com>
Date: Tue, 23 Apr 2024 20:42:47 -0700
Subject: [PATCH] add log dir

---
 src/baskerville/trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/baskerville/trainer.py b/src/baskerville/trainer.py
index 775fe64..74e6362 100644
--- a/src/baskerville/trainer.py
+++ b/src/baskerville/trainer.py
@@ -104,6 +104,7 @@ def __init__(
         if type(self.eval_data) is not list:
             self.eval_data = [self.eval_data]
         self.out_dir = out_dir
+        self.log_dir = log_dir
         self.strategy = strategy
         self.num_gpu = num_gpu
         self.batch_size = self.train_data[0].batch_size