Cian-H/symbolic_nn_tests
1
0
Fork 0
mirror of https://github.com/Cian-H/symbolic_nn_tests.git synced 2025-12-22 22:22:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update before first push to remote

Browse Source
This commit is contained in:
Cian Hughes
2024-05-15 15:03:51 +01:00
parent 15a57cc229
commit 4212c543f8
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
symbolic_nn_tests/__main__.py
View File

@@ -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)