Adding some suggests from the author

2023-02-14 22:20:30 +07:00
parent 721f8510af
commit 493671574e
5 changed files with 188 additions and 3 deletions
--- a/custom_nodes/example_folder/utils.py
+++ b/custom_nodes/example_folder/utils.py
@@ -0,0 +1,4 @@
+import torch
+def waste_cpu_resource():
+    x = torch.rand(1, 1e6, dtype=torch.float64).cpu()
+    return x.numpy()[0, 1]