(self)
| 70 | |
| 71 | class TestTensorFlow(unittest.TestCase): |
| 72 | def test_graph(self): |
| 73 | m = TrafficSignsModel() |
| 74 | |
| 75 | dot = hl.build_graph(m.graph).build_dot() |
| 76 | dot.format = 'pdf' |
| 77 | dot.render("tf_traffic_signs", directory=OUTPUT_DIR, cleanup=True) |
| 78 | |
| 79 | # Import CIFAR from the demos folder |
| 80 | sys.path.append("../demos") |
| 81 | import tf_cifar10 |
| 82 | |
| 83 | with tf.Session(): |
| 84 | with tf.Graph().as_default() as g: |
| 85 | tf_cifar10.CIFAR10(data_dir=DATA_DIR).model(inputs=tf.placeholder(tf.float32, shape=(8, 32, 32, 3))) |
| 86 | dot = hl.build_graph(g).build_dot() |
| 87 | dot.format = 'pdf' |
| 88 | dot.render("tf_cifar10", directory=OUTPUT_DIR, cleanup=True) |
| 89 | |
| 90 | |
| 91 | class TestSlimModels(unittest.TestCase): |
nothing calls this directly
no test coverage detected