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

Made loggers togglable from top level

Browse Source
This commit is contained in:
Cian Hughes
2024-05-23 09:21:48 +01:00
parent 1be0fa1020
commit 02b9e43f4d
1 changed files with 27 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
symbolic_nn_tests/__main__.py
View File

@@ -1,25 +1,36 @@
LEARNING_RATE = 10e-5 LEARNING_RATE = 10e-5
def qmnist_test(loss_func, version): def qmnist_test(loss_func, version, tensorboard=True, wandb=True):
from .experiment_1.model import main as test_model from .experiment_1.model import main as test_model
from lightning.pytorch.loggers import TensorBoardLogger
from lightning.pytorch.loggers import WandbLogger
import wandb
tb_logger = TensorBoardLogger( logger = []
save_dir=".",
name="logs/comparison", if tensorboard:
version=version, from lightning.pytorch.loggers import TensorBoardLogger
)
wandb_logger = WandbLogger( tb_logger = TensorBoardLogger(
project="Symbolic_NN_Tests", save_dir=".",
name=version, name="logs/comparison",
dir="wandb", version=version,
) )
logger = [tb_logger, wandb_logger] logger.append(tb_logger)
if wandb:
import wandb as _wandb
from lightning.pytorch.loggers import WandbLogger
wandb_logger = WandbLogger(
project="Symbolic_NN_Tests",
name=version,
dir="wandb",
)
logger.append(wandb_logger)
test_model(logger=logger, loss_func=loss_func, lr=LEARNING_RATE) test_model(logger=logger, loss_func=loss_func, lr=LEARNING_RATE)
wandb.finish()
if wandb:
_wandb.finish()
def qmnist_experiment(): def qmnist_experiment():