From 4376fb049b406434c20f233c3973a75d2f01b55e Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Tue, 26 Sep 2023 14:25:58 +0200 Subject: [PATCH] Comment out save_config statement in 02_train_tito --- examples/04_training/02_train_tito_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/04_training/02_train_tito_model.py b/examples/04_training/02_train_tito_model.py index 735dea055..f586b0581 100644 --- a/examples/04_training/02_train_tito_model.py +++ b/examples/04_training/02_train_tito_model.py @@ -182,7 +182,8 @@ def main( # Save model config and state dict - Version safe save method. model.save_state_dict(f"{path}/state_dict.pth") - model.save_config(f"{path}/model_config.yml") + # model.save_config(f"{path}/model_config.yml") + # Pending https://github.com/graphnet-team/graphnet/issues/606 if __name__ == "__main__":