From 406a8a277161547901476c7866e304036df078fd Mon Sep 17 00:00:00 2001 From: lmanan Date: Wed, 11 Oct 2023 21:45:20 -0400 Subject: [PATCH] Include initialize as config parameter --- cellulus/configs/model_config.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cellulus/configs/model_config.py b/cellulus/configs/model_config.py index 81c69c3..13dd74c 100644 --- a/cellulus/configs/model_config.py +++ b/cellulus/configs/model_config.py @@ -40,6 +40,9 @@ class ModelConfig: checkpoint will be used to resume training, otherwise the network will be trained from scratch. + initialize (default: False) + + If True, initialize the model weights with Kaiming Normal """ @@ -52,3 +55,4 @@ class ModelConfig: ] ) checkpoint: Path = attrs.field(default=None, converter=to_path) + initialize: bool = attrs.field(default=False, validator=instance_of(bool))