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)