Name	Size	Mode	Actions
seq2seq/	-	0755	rm
__pycache__/	-	0755	rm
download.py	8003	0644	editdlrm
imagenet_trainer_test_utils.py	5755	0644	editdlrm
resnet.py	11972	0644	editdlrm
resnet_test.py	2008	0644	editdlrm
shufflenet.py	7832	0644	editdlrm
shufflenet_test.py	2056	0644	editdlrm
__init__.py	0	0644	editdlrm
__sym_init__.py	494	0644	editdlrm

import os from caffe2.proto import caffe2_pb2 def _parseFile(filename): out_net = caffe2_pb2.NetDef() # TODO(bwasti): A more robust handler for pathnames. dir_path = os.path.dirname(__file__) with open('{dir_path}/{filename}'.format(dir_path=dir_path, filename=filename), 'rb') as f: out_net.ParseFromString(f.read()) return out_net init_net = _parseFile('init_net.pb') predict_net = _parseFile('predict_net.pb')