test: text log

SwanHubX · Feb 22, 2024 · d1d6f8f · d1d6f8f
1 parent c75c942
commit d1d6f8f
Showing 1 changed file with 32 additions and 22 deletions.
diff --git a/test/create_experiment.py b/test/create_experiment.py
@@ -19,28 +19,38 @@
     logggings=True,
 )
 for epoch in range(2, epochs):
-    if epoch % 10 == 0:
-        # # 测试audio
-        # sample_rate = 44100
-        # test_audio_arr = np.random.randn(2, 100000)
-        # swanlab.log(
-        #     {
-        #         "test/audio": [swanlab.Audio(test_audio_arr, sample_rate, caption="test")] * (epoch // 10),
-        #     },
-        #     step=epoch,
-        # )
-        # 测试image
-        test_image = np.random.randint(0, 255, (100, 100, 3))
-        swanlab.log(
-            {
-                # "test/image": swanlab.Image(test_image, caption="test"),
-                "test/text": swanlab.Text("this is a test text", caption="test"),
-            },
-            step=epoch,
-        )
-    acc = 1 - 2**-epoch - random.random() / epoch - offset
-    loss = 2**-epoch + random.random() / epoch + offset
-    loss2 = 3**-epoch + random.random() / epoch + offset * 3
+    # if epoch % 10 == 0:
+    # # 测试audio
+    # sample_rate = 44100
+    # test_audio_arr = np.random.randn(2, 100000)
+    # swanlab.log(
+    #     {
+    #         "test/audio": [swanlab.Audio(test_audio_arr, sample_rate, caption="test")] * (epoch // 10),
+    #     },
+    #     step=epoch,
+    # )
+    # 测试image
+    # test_image = np.random.randint(0, 255, (100, 100, 3))
+    # swanlab.log(
+    #     {
+    #         # "test/image": swanlab.Image(test_image, caption="test"),
+    #         "test/text": swanlab.Text("this is a test text", caption="test"),
+    #     },
+    #     step=epoch,
+    # )
+    # acc = 1 - 2**-epoch - random.random() / epoch - offset
+    # loss = 2**-epoch + random.random() / epoch + offset
+    # loss2 = 3**-epoch + random.random() / epoch + offset * 3
+    swanlab.log(
+        {
+            # "test/image": swanlab.Image(test_image, caption="test"),
+            "test/text": swanlab.Text(
+                "test test test test test test test test test test test test test test test",
+                caption="test test test test test test test test test test test test test test test",
+            ),
+        },
+        step=epoch,
+    )
     # print(f"epoch={epoch}, accuracy={acc}, loss={loss}")
     # swanlab.log({"t/accuracy": acc, "loss": loss, "loss2": loss2})
     time.sleep(2)