diff options
Diffstat (limited to 'python/caffe/test/test_net_spec.py')
-rw-r--r-- | python/caffe/test/test_net_spec.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/caffe/test/test_net_spec.py b/python/caffe/test/test_net_spec.py index 65b73b96..909a101b 100644 --- a/python/caffe/test/test_net_spec.py +++ b/python/caffe/test/test_net_spec.py @@ -43,7 +43,7 @@ def anon_lenet(batch_size): class TestNetSpec(unittest.TestCase): def load_net(self, net_proto): - f = tempfile.NamedTemporaryFile(delete=False) + f = tempfile.NamedTemporaryFile(mode='w+', delete=False) f.write(str(net_proto)) f.close() return caffe.Net(f.name, caffe.TEST) |