训练structure net遇到:
raceback (most recent call last):
File "./train_box.py", line 323, in <module>
train(config)
File "./train_box.py", line 101, in train
from tensorboardX import SummaryWriter
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/__init__.py", line 5, in <module>
from .torchvis import TorchVis
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/torchvis.py", line 11, in <module>
from .writer import SummaryWriter
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/writer.py", line 15, in <module>
from .event_file_writer import EventFileWriter
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/event_file_writer.py", line 28, in <module>
from .proto import event_pb2
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/proto/event_pb2.py", line 15, in <module>
from tensorboardX.proto import summary_pb2 as tensorboardX_dot_proto_dot_summary__pb2
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/proto/summary_pb2.py", line 15, in <module>
from tensorboardX.proto import tensor_pb2 as tensorboardX_dot_proto_dot_tensor__pb2
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/proto/tensor_pb2.py", line 15, in <module>
from tensorboardX.proto import resource_handle_pb2 as tensorboardX_dot_proto_dot_resource__handle__pb2
File "/home/s/anaconda2/envs/torch1.0/lib/python3.6/site-packages/tensorboardX/proto/resource_handle_pb2.py", line 22, in <module>
serialized_pb=_b('\n(tensorboardX/proto/resource_handle.proto\x12\x0ctensorboardX\"r\n\x13ResourceHandleProto\x12\x0e\n\x06\x64\x65vice\x18\x01 \x01(\t\x12\x11\n\tcontainer\x18\x02 \x01(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x11\n\thash_code\x18\x04 \x01(\x04\x12\x17\n\x0fmaybe_type_name\x18\x05 \x01(\tB/\n\x18org.tensorflow.frameworkB\x0eResourceHandleP\x01\xf8\x01\x01\x62\x06proto3')
google一下是protobuf版本问题。
protoc --version 显示
libprotoc 3.4.0
解决方法:
pip install -U protobuf
升级到protobuf-3.9.1即可