Add directory for flame graphs and add bash scripts

wentongli · Oct 5, 2020 · 0eaec9c · 0eaec9c
1 parent 88287c7
commit 0eaec9c
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/chapter7/bad_random.py → chapter7/flame_graphs/bad_random.py b/chapter7/bad_random.py → chapter7/flame_graphs/bad_random.py
diff --git a/chapter7/flame_graphs/create_flame_graph_bad_random.sh b/chapter7/flame_graphs/create_flame_graph_bad_random.sh
@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+
+py-spy record -r 99 -d 30 -o badrandom.svg -- python bad_random.py
diff --git a/chapter7/flame_graphs/create_flame_graph_good_random.sh b/chapter7/flame_graphs/create_flame_graph_good_random.sh
@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+
+py-spy record -r 99 -d 30 -o goodrandom.svg -- python good_random.py
diff --git a/chapter7/good_random.py → chapter7/flame_graphs/good_random.py b/chapter7/good_random.py → chapter7/flame_graphs/good_random.py
@@ -5,8 +5,7 @@
 from torch.utils.tensorboard import SummaryWriter
 from torchvision import datasets, transforms,models
 import torch.utils.data
-from PIL import Image
-import numpy as np
+
 
 model = models.resnet18(pretrained=True)
 device = "cuda:0"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		#!/usr/bin/env bash

		py-spy record -r 99 -d 30 -o badrandom.svg -- python bad_random.py
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		#!/usr/bin/env bash

		py-spy record -r 99 -d 30 -o goodrandom.svg -- python good_random.py