v1.7.0

TylerYep · May 28, 2022 · 9edffed · 9edffed
1 parent cf8cb6e
commit 9edffed
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/profile.txt b/profile.txt
diff --git a/profiler.py b/profiler.py
@@ -26,7 +26,7 @@ def profile() -> None:
     """
     random.seed(0)
     command = (
-        "for _ in trange(50): "
+        "for _ in trange(10): "
         "summary(torchvision.models.resnet152(), (1, 3, 224, 224), verbose=0)"
     )
     profile_file = "profile.txt"