From 147c156c8a2ae6b5c0a1cc6c0a0cb4bd78cfceae Mon Sep 17 00:00:00 2001
From: Neil Kim Nielsen <37108154+neilkimn@users.noreply.github.com>
Date: Wed, 4 Oct 2023 18:38:30 +0200
Subject: [PATCH] Make TrackerLoadConfig optional (#306)

---
 dalle2_pytorch/train_configs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dalle2_pytorch/train_configs.py b/dalle2_pytorch/train_configs.py
index 00a4225..ecdf994 100644
--- a/dalle2_pytorch/train_configs.py
+++ b/dalle2_pytorch/train_configs.py
@@ -90,7 +90,7 @@ class TrackerConfig(BaseModel):
     data_path: str = '.tracker_data'
     overwrite_data_path: bool = False
     log: TrackerLogConfig
-    load: Optional[TrackerLoadConfig]
+    load: Optional[TrackerLoadConfig] = None
     save: Union[List[TrackerSaveConfig], TrackerSaveConfig]
 
     def create(self, full_config: BaseModel, extra_config: dict, dummy_mode: bool = False) -> Tracker: