| from __future__ import absolute_import |
| from __future__ import division |
| from __future__ import print_function |
| from __future__ import unicode_literals |
| from caffe2.python import core, workspace |
| example_files = glob.glob('example_*.c2s') |
| print('Running example file', ex) |
| inits = json.loads(f.readline()) |
| net_name = f.readline().strip() |
| outputs = json.loads(f.readline()) |
| CU = core.C.CompilationUnit() |
| # Initialize workspace with required inputs |
| for name, shape, dt in inits: |
| workspace.FeedBlob(name, np.random.rand(*shape).astype(np.dtype(dt))) |
| net = CU.create_net(net_name) |
| print('Success! Interesting outputs:') |
| print(output, workspace.FetchBlob(output)) |