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__":