Dev icenet-ai#252: couple of notes

JimCircadian · Jun 7, 2024 · 7a519cd · 7a519cd
1 parent 6bd2cfa
commit 7a519cd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/icenet/model/networks/tensorflow.py b/icenet/model/networks/tensorflow.py
@@ -76,6 +76,8 @@ def train(self,
         with strategy.scope():
             network = model_creator(**model_creator_kwargs)
 
+        # TODO: consider using .keras format throughout
+        # TODO: need to consider pre_load / create and save functionality for checkpoint recovery
         if self._pre_load_path and os.path.exists(self._pre_load_path):
             logging.warning("Automagically loading network weights from {}".format(
                 self._pre_load_path))