From 4212c543f8af768d7d4a028fa48a3f9469e0ddb9 Mon Sep 17 00:00:00 2001 From: Cian-H Date: Wed, 15 May 2024 15:03:51 +0100 Subject: [PATCH] Update before first push to remote --- README.md | 4 ++++ symbolic_nn_tests/__main__.py | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index e69de29..ad1d386 100644 --- a/README.md +++ b/README.md @@ -0,0 +1,4 @@ +A sandbox for testing simple ways of injecting symbolic knowledge into neural networks. + +This sandbox currently tests: +- Semantic loss functions diff --git a/symbolic_nn_tests/__main__.py b/symbolic_nn_tests/__main__.py index f308fc5..1a88eb1 100644 --- a/symbolic_nn_tests/__main__.py +++ b/symbolic_nn_tests/__main__.py @@ -11,7 +11,7 @@ def run_test(loss_func, version): name="logs/comparison", version=version, ) - wandb_logger = WandbLogger(project="MNIST") + wandb_logger = WandbLogger(project="Semantic_Loss_Tests") logger = [tb_logger, wandb_logger] test_model(logger=logger, loss_func=loss_func, lr=LEARNING_RATE)