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

Added wandb logging

Browse Source
This commit is contained in:
Cian Hughes
2024-05-15 13:37:29 +01:00
parent 6600a79f71
commit 15a57cc229
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -164,3 +164,4 @@ cython_debug/
datasets/
lightning_logs/
logs/
wandb/

16
symbolic_nn_tests/__main__.py
View File

@@ -4,14 +4,16 @@ LEARNING_RATE = 10e-5
def run_test(loss_func, version):
from .model import main as test_model
from lightning.pytorch.loggers import TensorBoardLogger
from lightning.pytorch.loggers import WandbLogger
logger = TensorBoardLogger(
tb_logger = TensorBoardLogger(
save_dir=".",
name="logs/comparison",
version=version,
)
test_model(lr=LEARNING_RATE)
# test_model(logger=logger, loss_func=loss_func, lr=LEARNING_RATE)
wandb_logger = WandbLogger(project="MNIST")
logger = [tb_logger, wandb_logger]
test_model(logger=logger, loss_func=loss_func, lr=LEARNING_RATE)
def main():
@@ -19,10 +21,10 @@ def main():
from torch import nn
run_test(nn.functional.cross_entropy, "cross_entropy")
# run_test(semantic_loss.similarity_cross_entropy, "similarity_cross_entropy")
# run_test(semantic_loss.hasline_cross_entropy, "hasline_cross_entropy")
# run_test(semantic_loss.hasloop_cross_entropy, "hasloop_cross_entropy")
# run_test(semantic_loss.multisemantic_cross_entropy, "multisemantic_cross_entropy")
run_test(semantic_loss.similarity_cross_entropy, "similarity_cross_entropy")
run_test(semantic_loss.hasline_cross_entropy, "hasline_cross_entropy")
run_test(semantic_loss.hasloop_cross_entropy, "hasloop_cross_entropy")
run_test(semantic_loss.multisemantic_cross_entropy, "multisemantic_cross_entropy")
if __name__ == "__main__":