diff --git a/profile.txt b/profile.txt deleted file mode 100644 index b66d65e..0000000 Binary files a/profile.txt and /dev/null differ diff --git a/profiler.py b/profiler.py index cd5a6db..1ebace4 100644 --- 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"