提交 917e41fd 编写于 作者: S superjom

update some test

上级 7f500f45
...@@ -119,7 +119,6 @@ if __name__ == '__main__': ...@@ -119,7 +119,6 @@ if __name__ == '__main__':
tags = get_image_tags(reader, 'train') tags = get_image_tags(reader, 'train')
tags = get_image_tag_steps(reader, 'train', 'layer1/layer2/image0/0') tags = get_image_tag_steps(reader, 'train', 'layer1/layer2/image0/0')
print 'image step tags'
pprint.pprint(tags) pprint.pprint(tags)
image = get_invididual_image(reader, "train", 'layer1/layer2/image0/0', 2) image = get_invididual_image(reader, "train", 'layer1/layer2/image0/0', 2)
......
...@@ -21,7 +21,6 @@ test_scalar1 = test_writer.scalar("model/scalar/max") ...@@ -21,7 +21,6 @@ test_scalar1 = test_writer.scalar("model/scalar/max")
for i in range(100): for i in range(100):
train_scalar.add_record(i, random.random()) train_scalar.add_record(i, random.random())
train_scalar1.add_record(i, random.random()) train_scalar1.add_record(i, random.random())
#time.sleep(1)
if i % 10 == 0: if i % 10 == 0:
test_scalar.add_record(i, random.random()) test_scalar.add_record(i, random.random())
test_scalar1.add_record(i, random.random()) test_scalar1.add_record(i, random.random())
...@@ -31,13 +30,14 @@ def add_image(mode): ...@@ -31,13 +30,14 @@ def add_image(mode):
tag = "layer1/layer2/image0" tag = "layer1/layer2/image0"
# TODO check step_cycle # TODO check step_cycle
num_samples = 10 num_samples = 10
num_passes = 10 num_passes = 20
image_writer = writer_.image(tag, num_samples, 1) image_writer = writer_.image(tag, num_samples, 1)
shape = [400, 400, 3] shape = [50, 50, 3]
for pass_ in xrange(num_passes): for pass_ in xrange(num_passes):
image_writer.start_sampling() image_writer.start_sampling()
for ins in xrange(2*num_samples): for ins in xrange(2*num_samples):
print '.',
index = image_writer.is_sample_taken() index = image_writer.is_sample_taken()
if index != -1: if index != -1:
data = np.random.random(shape) * 256 data = np.random.random(shape) * 256
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册