From 15ede5d085474431913f3d7001b413dff5d09007 Mon Sep 17 00:00:00 2001 From: Katia Date: Mon, 15 Aug 2022 23:14:26 -0700 Subject: [PATCH] use wandb.watch --- tests/unit_tests/test_torch_full.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit_tests/test_torch_full.py b/tests/unit_tests/test_torch_full.py index de3c9381c88..44b6a217768 100644 --- a/tests/unit_tests/test_torch_full.py +++ b/tests/unit_tests/test_torch_full.py @@ -160,7 +160,7 @@ def test_all_logging(relay_server, wandb_init): with relay_server() as relay: run = wandb_init() net = ConvNet() - run.watch( + wandb.watch( net, log="all", log_freq=1, @@ -188,7 +188,7 @@ def test_embedding_dict_watch(relay_server, wandb_init): with relay_server() as relay: run = wandb_init() model = EmbModelWrapper() - run.watch(model, log_freq=1, idx=0) + wandb.watch(model, log_freq=1, idx=0) opt = torch.optim.Adam(params=model.parameters()) inp = torch.randint(16, [8, 5]) out = model(inp)