We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent cf8cb6e commit 9edffedCopy full SHA for 9edffed
2 files changed
profile.txt
-64.6 KB
profiler.py
@@ -26,7 +26,7 @@ def profile() -> None:
26
"""
27
random.seed(0)
28
command = (
29
- "for _ in trange(50): "
+ "for _ in trange(10): "
30
"summary(torchvision.models.resnet152(), (1, 3, 224, 224), verbose=0)"
31
)
32
profile_file = "profile.txt"
0 commit comments