summaryrefslogtreecommitdiff
path: root/python/caffe/test/test_net_spec.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/caffe/test/test_net_spec.py')
-rw-r--r--python/caffe/test/test_net_spec.py2
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)